starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #111 from QwenLM/update_cli_demo

Browse Source 
Update cli_demo.py
This commit is contained in:
Yang An
2023-08-07 22:57:19 +08:00
committed by GitHub
parent 2c0761afe7 982eef010d
commit 3531385d0c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cli_demo.py
View File

@@ -177,7 +177,7 @@ def main():
# Run chat. # Run chat.
set_seed(seed) set_seed(seed)
try: try:
for response in model.chat(tokenizer, query, history=history, stream=True): for response in model.chat_stream(tokenizer, query, history=history):
_clear_screen() _clear_screen()
print(f"\nUser: {query}") print(f"\nUser: {query}")
print(f"\nQwen-7B: {response}") print(f"\nQwen-7B: {response}")