examples : fix memory leak on failure to load gpt2 model (ggerganov#323)

mindkhichdi · Dec 23, 2022 · 331c0bb · 331c0bb
1 parent dc90efd
commit 331c0bb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/talk/gpt-2.cpp b/examples/talk/gpt-2.cpp
@@ -841,6 +841,7 @@ struct gpt2_context * gpt2_init(const char * path_model) {
 
         if (!gpt2_model_load(path_model, ctx->model, ctx->vocab)) {
             fprintf(stderr, "%s: failed to load model from '%s'\n", __func__, path_model);
+            delete ctx;
             return nullptr;
         }