diff --git a/qwen/research_attention.py b/qwen/research_attention.py index 3ab98c3..f0a42d8 100644 --- a/qwen/research_attention.py +++ b/qwen/research_attention.py @@ -62,7 +62,7 @@ class ResearchRunner(QwenRunner): runner = ResearchRunner(model) # 第一轮对话 -response, history, decode_tokens = runner.Chat(tokenizer, "东南亚国家日本的首都是什么市", "") +response, history, decode_tokens = runner.Chat(tokenizer, "东南亚国家日本的首都是什么市", "日本的首都是") print(decode_tokens) # <|im_start|>system # You are a helpful assistant.<|im_end|> @@ -71,6 +71,6 @@ print(decode_tokens) # <|im_start|>assistant # 日本的首都东京。<|im_end|> # <|endoftext|> - + # 日本的首都是东京。<|im_end|> if decode_tokens.split("\n")[-2] != """日本的首都东京。<|im_end|>""": raise ()