diff --git a/chat/web_demo.py b/chat/web_demo.py index 5221237..93272a6 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -161,11 +161,11 @@ def on_btn_click(): @st.cache_resource def load_model(): model = ( - AutoModelForCausalLM.from_pretrained("/mnt/inspurfs/share_data/zhangwenwei/models/hf_release/internlm2-chat-7b", trust_remote_code=True) + AutoModelForCausalLM.from_pretrained("internlm/internlm2-chat-7b", trust_remote_code=True) .to(torch.bfloat16) .cuda() ) - tokenizer = AutoTokenizer.from_pretrained("/mnt/inspurfs/share_data/zhangwenwei/models/hf_release/internlm2-chat-7b", trust_remote_code=True) + tokenizer = AutoTokenizer.from_pretrained("internlm/internlm2-chat-7b", trust_remote_code=True) return model, tokenizer