diff --git a/web_demo.py b/web_demo.py index 922df00..11d7a56 100755 --- 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():