fb700 commited on
Commit
e39ff0f
1 Parent(s): c5469ad

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -25,8 +25,6 @@ RETRY_FLAG = False
25
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
26
  #model = AutoModel.from_pretrained(model_name, trust_remote_code=True).quantize(4).half().cuda()
27
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True).half().cuda()
28
- from peft import PeftModel
29
- model = PeftModel.from_pretrained(model, 'D:\glm\model\lora\Generalization_rlhf')
30
  model = model.eval()
31
 
32
  _ = """Override Chatbot.postprocess"""
 
25
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
26
  #model = AutoModel.from_pretrained(model_name, trust_remote_code=True).quantize(4).half().cuda()
27
  model = AutoModel.from_pretrained(model_name, trust_remote_code=True).half().cuda()
 
 
28
  model = model.eval()
29
 
30
  _ = """Override Chatbot.postprocess"""