Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
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"""
|