Hugging Face's logo

Spaces:
ljsabc
/
Fujisaki
Runtime error

App Files Community
1
ljsabc commited on
Commit
e0c2263
•
1 Parent(s): 050e044

Refactor to use upstream automodel.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import torch
2
 
3
  from transformers import AutoTokenizer, GenerationConfig, AutoModel
4
 
5
- model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, revision="fdb7a60").float()
6
- tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, revision="fdb7a60")
7
 
8
  from peft import PeftModel
9
  peft_path = 'ljsabc/Fujisaki_GLM' # change it to your own
 
2
 
3
  from transformers import AutoTokenizer, GenerationConfig, AutoModel
4
 
5
+ model = AutoModel.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, revision="658202d").float()
6
+ tokenizer = AutoTokenizer.from_pretrained("THUDM/chatglm-6b", trust_remote_code=True, revision="658202d")
7
 
8
  from peft import PeftModel
9
  peft_path = 'ljsabc/Fujisaki_GLM' # change it to your own