From 34f2f5aa96279ded048124f50f7b1cc58f104eee Mon Sep 17 00:00:00 2001 From: ZwwWayne Date: Thu, 18 Jan 2024 22:07:03 +0800 Subject: [PATCH] update web demo --- chat/web_demo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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