embedding : free the batch after execution (ggerganov#7297)

teleprint-me · May 15, 2024 · ea3b059 · ea3b059
1 parent 29499bb
commit ea3b059
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/embedding/embedding.cpp b/examples/embedding/embedding.cpp
@@ -211,6 +211,7 @@ int main(int argc, char ** argv) {
 
     // clean up
     llama_print_timings(ctx);
+    llama_batch_free(batch);
     llama_free(ctx);
     llama_free_model(model);
     llama_backend_free();