From fe6dc611430e90f2d4c02c7a8264a3a11a74464c Mon Sep 17 00:00:00 2001
From: gtygo <gtydoit@gmail.com>
Date: Sat, 10 Aug 2024 01:11:51 +0800
Subject: [PATCH] retrieval

---
 examples/retrieval/retrieval.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/retrieval/retrieval.cpp b/examples/retrieval/retrieval.cpp
index 65b19ce71cbe3..8628f4c281e75 100644
--- a/examples/retrieval/retrieval.cpp
+++ b/examples/retrieval/retrieval.cpp
@@ -266,7 +266,7 @@ int main(int argc, char ** argv) {
         std::vector<float> query_emb(n_embd, 0);
         batch_decode(ctx, query_batch, query_emb.data(), 1, n_embd);
 
-        llama_batch_clear(query_batch);
+        llama_batch_free(query_batch);
 
         // compute cosine similarities
         {