Kaynağa Gözat

embedding : free the batch after execution (#7297)

dm4 1 yıl önce
ebeveyn
işleme
ea3b0590ee
1 değiştirilmiş dosya ile 1 ekleme ve 0 silme
  1. 1 0
      examples/embedding/embedding.cpp

+ 1 - 0
examples/embedding/embedding.cpp

@@ -211,6 +211,7 @@ int main(int argc, char ** argv) {
 
 
     // clean up
     // clean up
     llama_print_timings(ctx);
     llama_print_timings(ctx);
+    llama_batch_free(batch);
     llama_free(ctx);
     llama_free(ctx);
     llama_free_model(model);
     llama_free_model(model);
     llama_backend_free();
     llama_backend_free();