Merge pull request #247 from QwenLM/update_web_demo

Update web demo
1 year ago · 80de517162
parent f56550b01f 09f189449c
commit 80de517162
1 changed files with 2 additions and 1 deletions
--- a/web_demo.py
+++ b/web_demo.py
@ -14,7 +14,8 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation import GenerationConfig

-DEFAULT_CKPT_PATH = 'QWen/QWen-7B-Chat'
+
+DEFAULT_CKPT_PATH = 'Qwen/Qwen-7B-Chat'


 def _get_args():