typo fixed: ChatGLM -> ChatGLM2

git000 · Jun 26, 2023 · 55d7548 · 55d7548
1 parent 07957ba
commit 55d7548
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cli_demo.py b/cli_demo.py
@@ -14,10 +14,10 @@
 
 
 def build_prompt(history):
-    prompt = "欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序"
+    prompt = "欢迎使用 ChatGLM2-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序"
     for query, response in history:
         prompt += f"\n\n用户：{query}"
-        prompt += f"\n\nChatGLM-6B：{response}"
+        prompt += f"\n\nChatGLM2-6B：{response}"
     return prompt