github
/
ChatGLM-6B
mirror of https://github.com/THUDM/ChatGLM-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add prompt after clear

pull/41/head
duzx16 2023-03-14 20:39:55 +08:00
parent afee32f2b1
commit fc4ac83501
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
cli_demo.py
View File

@ -17,7 +17,8 @@ while True:
if query == "clear": if query == "clear":
history = [] history = []
command = 'cls' if os_name == 'Windows' else 'clear' command = 'cls' if os_name == 'Windows' else 'clear'
os.system(command) os.system(command)
print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序")
continue continue
response, history = model.chat(tokenizer, query, history=history) response, history = model.chat(tokenizer, query, history=history)
print(f"ChatGLM-6B：{response}") print(f"ChatGLM-6B：{response}")