From a1fd0b8750dd086e490994e39329217aea2f8922 Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Thu, 25 May 2023 22:00:17 +0800 Subject: [PATCH] Update chatglm_llm.py --- models/chatglm_llm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/chatglm_llm.py b/models/chatglm_llm.py index ea537370..5d440557 100644 --- a/models/chatglm_llm.py +++ b/models/chatglm_llm.py @@ -63,7 +63,7 @@ class ChatGLM(BaseAnswer, LLM, ABC): temperature=self.temperature, stopping_criteria=stopping_criteria_list )): - self.checkPoint.clear_torch_cache() +# self.checkPoint.clear_torch_cache() history[-1] = [prompt, stream_resp] answer_result = AnswerResult() answer_result.history = history