LoftQ commited on
Commit
d9c5270
1 Parent(s): 2c16204

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ from peft import PeftModel
40
 
41
  MODEL_ID = "LoftQ/Phi-3-mini-4k-instruct-4bit-64rank"
42
 
43
- base_model = AutoModelForCausalLM.from_pretrained(MODEL_ID)
44
  peft_model = PeftModel.from_pretrained(
45
  base_model,
46
  MODEL_ID,
 
40
 
41
  MODEL_ID = "LoftQ/Phi-3-mini-4k-instruct-4bit-64rank"
42
 
43
+ base_model = AutoModelForCausalLM.from_pretrained(MODEL_ID, trust_remote_code=True)
44
  peft_model = PeftModel.from_pretrained(
45
  base_model,
46
  MODEL_ID,