diff --git a/web_demo.py b/web_demo.py index 0a0914a..11d7a56 100755 --- a/web_demo.py +++ b/web_demo.py @@ -14,6 +14,7 @@ import torch from transformers import AutoModelForCausalLM, AutoTokenizer from transformers.generation import GenerationConfig + DEFAULT_CKPT_PATH = 'Qwen/Qwen-7B-Chat'