darshjoshi16 commited on
Commit
d7d6de0
·
verified ·
1 Parent(s): 9a00b32

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -38,10 +38,10 @@ from peft import PeftModel
38
 
39
  # Load base model and tokenizer
40
  base_model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2")
41
- tokenizer = AutoTokenizer.from_pretrained("your-username/phi2-lora-math")
42
 
43
  # Load LoRA adapter
44
- model = PeftModel.from_pretrained(base_model, "your-username/phi2-lora-math")
45
 
46
  # Inference
47
  prompt = "Q: Julie read 12 pages yesterday and twice as many today. If she wants to read half of the remaining 84 pages tomorrow, how many pages should she read?\nA:"
 
38
 
39
  # Load base model and tokenizer
40
  base_model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2")
41
+ tokenizer = AutoTokenizer.from_pretrained("darshjoshi16/phi2-lora-math")
42
 
43
  # Load LoRA adapter
44
+ model = PeftModel.from_pretrained(base_model, "darshjoshi16/phi2-lora-math")
45
 
46
  # Inference
47
  prompt = "Q: Julie read 12 pages yesterday and twice as many today. If she wants to read half of the remaining 84 pages tomorrow, how many pages should she read?\nA:"