bugfix of crash

wangzhaode · Jan 9, 2024 · 67d21e1 · 67d21e1
1 parent f66fe6a
commit 67d21e1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/include/llm.hpp b/include/llm.hpp
@@ -52,7 +52,11 @@ class Llm {
         // default tokenier is senrencepiece
         tokenizer_.reset(new Sentencepiece);
     }
-    virtual ~Llm() = default;
+    virtual ~Llm() {
+        modules_.clear();
+        visual_module_.reset();
+        runtime_manager_.reset();
+    }
     static Llm* createLLM(const std::string& path, std::string model_type = "auto");
     void load(const std::string& model_dir);
     void chat();