Spaces:

MaximeSHE
/

chatbot

Sleeping

MaximeSHE commited on Sep 26, 2024

Commit

33c1b73

verified ·

1 Parent(s): 78fd0dc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -181,7 +181,7 @@ def on_btn_click():
 @st.cache_resource
 def load_model():
-    model_dir = '/home/scy/models/Shanghai_AI_Laboratory/internlm2-chat-1_8b'
     model = (AutoModelForCausalLM.from_pretrained(
         model_dir,
         trust_remote_code=True).to(torch.bfloat16).cuda())

 @st.cache_resource
 def load_model():
+    model_dir = 'internlm/internlm2-chat-1_8b'
     model = (AutoModelForCausalLM.from_pretrained(
         model_dir,
         trust_remote_code=True).to(torch.bfloat16).cuda())