Update model path

tiansher123 · Apr 6, 2023 · 40d83f3 · 40d83f3
1 parent cc4be39
commit 40d83f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/web_demo3.py b/web_demo3.py
@@ -2,8 +2,8 @@
 import gradio as gr
 import mdtex2html
 
-tokenizer = AutoTokenizer.from_pretrained("/mnt/vepfs/workspace/zxdu/chatglm_6b", trust_remote_code=True)
-model = AutoModel.from_pretrained("/mnt/vepfs/workspace/zxdu/chatglm_6b", trust_remote_code=True).half().cuda()
+tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True)
+model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True).half().cuda()
 model = model.eval()
 
 """Override Chatbot.postprocess"""