From 982eef010d148bd595d0c9296b22307c2d9306ad Mon Sep 17 00:00:00 2001 From: Yang An Date: Mon, 7 Aug 2023 22:57:07 +0800 Subject: [PATCH] Update cli_demo.py --- cli_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cli_demo.py b/cli_demo.py index 3e97735..52a3a63 100644 --- a/cli_demo.py +++ b/cli_demo.py @@ -177,7 +177,7 @@ def main(): # Run chat. set_seed(seed) try: - for response in model.chat(tokenizer, query, history=history, stream=True): + for response in model.chat_stream(tokenizer, query, history=history): _clear_screen() print(f"\nUser: {query}") print(f"\nQwen-7B: {response}")