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

chore(cli_demo): strip the query input for `stop` or `clear` command 

Remove the query space to determine whether it is a stop command or a clear command
pull/322/head
Ashing Zheng 2023-04-01 13:18:28 +08:00 committed by GitHub
parent ff3761fc1a
commit 6dd6f7c7b5
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 终止程序")