embedding: free the batch after execution

2024-05-15 17:14:44 +08:00 · 2024-05-15 17:14:44 +08:00 · 518fbc1f74
commit 518fbc1f74
parent 583fd6b000
1 changed files with 1 additions and 0 deletions
--- 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();