Gaze/ChatGLM2-6B
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix prompt in cli demo

Browse Source
This commit is contained in:
duzx16 2023-06-26 16:49:56 +08:00
parent 9d439304d4
commit 385040e6c7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cli_demo.py
View File

@ -14,7 +14,7 @@ stop_stream = False
def build_prompt(history): def build_prompt(history):
prompt = "欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序" prompt = "欢迎使用 ChatGLM2-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序"
for query, response in history: for query, response in history:
prompt += f"\n\n用户：{query}" prompt += f"\n\n用户：{query}"
prompt += f"\n\nChatGLM-6B：{response}" prompt += f"\n\nChatGLM-6B：{response}"