LeeKinXUn commited on
Commit
e997932
โ€ข
1 Parent(s): cf1c491

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from gradio.components import Textbox
4
 
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b-int4", trust_remote_code=True)
7
- model = AutoModel.from_pretrained("THUDM/chatglm-6b-int4", trust_remote_code=True).half().cuda()
8
  model.eval()
9
 
10
  history=[]
 
4
 
5
 
6
  tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b-int4", trust_remote_code=True)
7
+ model = AutoModel.from_pretrained("THUDM/chatglm-6b-int4", trust_remote_code=True).float()
8
  model.eval()
9
 
10
  history=[]