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

Merge pull request #322 from ronething/patch-1 

chore(cli_demo): strip the query input for `stop` or `clear` command
pull/328/head
Zhengxiao Du 2023-04-01 19:45:14 +08:00 committed by GitHub
parent ff3761fc1a 6dd6f7c7b5
commit 534e5f1a6e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cli_demo.py
View File

@ -31,9 +31,9 @@ def main():
print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序") print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序")
while True: while True:
query = input("\n用户：") query = input("\n用户：")
if query == "stop": if query.strip() == "stop":
break break
if query == "clear": if query.strip() == "clear":
history = [] history = []
os.system(clear_command) os.system(clear_command)
print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序") print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序")