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 #448 from JianxinMa/main

Browse Source 
fix badcase in react_demo.py
This commit is contained in:
Jianxin Ma
2023-10-11 17:40:06 +08:00
committed by GitHub
parent dbc111267a 8da5dc0cbd
commit 7a0d45de09
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
examples/react_demo.py
View File

@@ -18,13 +18,13 @@ from transformers.generation import GenerationConfig
for _ in range(10): # 网络不稳定，多试几次 for _ in range(10): # 网络不稳定，多试几次
try: try:
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True) name = 'Qwen/Qwen-7B-Chat'
generation_config = GenerationConfig.from_pretrained("Qwen/Qwen-7B-Chat", trust_remote_code=True) tokenizer = AutoTokenizer.from_pretrained(name, trust_remote_code=True)
generation_config = GenerationConfig.from_pretrained(name, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained( model = AutoModelForCausalLM.from_pretrained(
"Qwen/Qwen-7B-Chat", device_map="auto", trust_remote_code=True name, device_map="auto", trust_remote_code=True
).eval() ).eval()
model.generation_config = generation_config model.generation_config = generation_config
model.generation_config.do_sample = False
break break
except Exception: except Exception:
pass pass
@@ -236,7 +236,7 @@ def test():
}, },
] ]
history = [] history = []
for query in ['你好', '谁是周杰伦', '他老婆是谁', '给我画个可爱的小猫吧，最好是黑猫']: for query in ['你好', '查一下谁是周杰伦', '搜下他老婆是谁', '给我画个可爱的小猫吧，最好是黑猫']:
print(f"User's Query:\n{query}\n") print(f"User's Query:\n{query}\n")
response, history = llm_with_plugin(prompt=query, history=history, list_of_plugin_info=tools) response, history = llm_with_plugin(prompt=query, history=history, list_of_plugin_info=tools)
print(f"Qwen's Response:\n{response}\n") print(f"Qwen's Response:\n{response}\n")
@@ -254,7 +254,7 @@ Thought: 提供的工具对回答该问题帮助较小，我将不使用工具
Final Answer: 你好！很高兴见到你。有什么我可以帮忙的吗？ Final Answer: 你好！很高兴见到你。有什么我可以帮忙的吗？
User's Query: User's Query:
谁是周杰伦 查一下谁是周杰伦
Qwen's Response: Qwen's Response:
Thought: 我应该使用Google搜索查找相关信息。 Thought: 我应该使用Google搜索查找相关信息。
@@ -265,7 +265,7 @@ Thought: I now know the final answer.
Final Answer: 周杰伦（Jay Chou）是一位来自台湾的歌手、词曲创作人、音乐制作人、说唱歌手、演员、电视节目主持人和企业家。他以其独特的音乐风格和才华在华语乐坛享有很高的声誉。 Final Answer: 周杰伦（Jay Chou）是一位来自台湾的歌手、词曲创作人、音乐制作人、说唱歌手、演员、电视节目主持人和企业家。他以其独特的音乐风格和才华在华语乐坛享有很高的声誉。
User's Query: User's Query:
他老婆是谁 搜下他老婆是谁
Qwen's Response: Qwen's Response:
Thought: 我应该使用Google搜索查找相关信息。 Thought: 我应该使用Google搜索查找相关信息。