IntelLabs
/

shears-mpt-7b-50-gsm8k-super-adapter

Model card Files Files and versions

jinjieyuan commited on Mar 19

Commit

2a64565

•

1 Parent(s): e3526ec

Update README.md

Signed-off-by: jinjieyuan <[email protected]>

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -60,7 +60,7 @@ def generate_prompt(instruction):
 base_model_path = "shears-mpt-7b-50-gsm8k-super/base_model"
 adapter_model_path = "shears-mpt-7b-50-gsm8k-super/adapter_model"
-base_model = AutoModelForCausalLM.from_pretrained(base_model_path)
 model = PeftModel.from_pretrained(base_model, adapter_model_path)
 model.eval()

 base_model_path = "shears-mpt-7b-50-gsm8k-super/base_model"
 adapter_model_path = "shears-mpt-7b-50-gsm8k-super/adapter_model"
+base_model = AutoModelForCausalLM.from_pretrained(base_model_path, trust_remote_code=True)
 model = PeftModel.from_pretrained(base_model, adapter_model_path)
 model.eval()