retrieval

gtygo · Aug 9, 2024 · fe6dc61 · fe6dc61
1 parent 6afd1a9
commit fe6dc61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
         {