Просмотр исходного кода

embedding : free the batch after execution (#7297)

dm4 1 год назад
Родитель
Сommit
ea3b0590ee
1 измененных файлов с 1 добавлено и 0 удалено
  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
     llama_print_timings(ctx);
+    llama_batch_free(batch);
     llama_free(ctx);
     llama_free_model(model);
     llama_backend_free();