Spaces:

xiangxiang
/

chatglm2-6b

Running

xiangxiang commited on Aug 5, 2023

Commit

34eb64a

•

1 Parent(s): ca874bf

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,8 @@ from transformers import AutoModel, AutoTokenizer
 import gradio as gr
 import mdtex2html
-tokenizer = AutoTokenizer.from_pretrained("THUDM/ChatGLM-6B", trust_remote_code=True)
-model = AutoModel.from_pretrained("THUDM/ChatGLM-6B",trust_remote_code=True)
 model = model.eval()
 """Override Chatbot.postprocess"""

 import gradio as gr
 import mdtex2html
+tokenizer = AutoTokenizer.from_pretrained("THUDM/ChatGLM2-6B", trust_remote_code=True)
+model = AutoModel.from_pretrained("THUDM/ChatGLM2-6B",trust_remote_code=True).float()
 model = model.eval()
 """Override Chatbot.postprocess"""