From 5dbac40925220bf6dd249cb02f5ef088f8bbc147 Mon Sep 17 00:00:00 2001
From: Colin <colin>
Date: Sun, 21 Jan 2024 22:43:16 +0800
Subject: [PATCH] Refien.

---
 qwen/research_attention.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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 ()