Browse Source

Fix prompt in cli demo

pull/23/head
duzx16 1 year ago
parent
commit
385040e6c7
  1. 2
      cli_demo.py

2
cli_demo.py

@ -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}"

Loading…
Cancel
Save