Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
from transformers import AutoModel, AutoTokenizer
|
2 |
import gradio as gr
|
3 |
import json
|
4 |
-
model_path = 'THUDM/
|
5 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
6 |
model = AutoModel.from_pretrained(model_path, trust_remote_code=True).half().float()
|
7 |
model = model.eval()
|
|
|
1 |
from transformers import AutoModel, AutoTokenizer
|
2 |
import gradio as gr
|
3 |
import json
|
4 |
+
model_path = 'THUDM/chatglm2-6b-int4'
|
5 |
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)
|
6 |
model = AutoModel.from_pretrained(model_path, trust_remote_code=True).half().float()
|
7 |
model = model.eval()
|