reasonwang commited on
Commit
2dd5eef
·
1 Parent(s): 295ca87

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -14,7 +14,7 @@ from peft import PeftModel
14
  base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
15
  peft_model = PeftModel.from_pretrained(base_model,"reasonwang/flan-alpaca-lora-base")
16
 
17
- inputs = tokenizer("What should I say when my friend is crying?", return_tensors="pt")
18
  outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)
19
  print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
20
  ```
 
14
  base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
15
  peft_model = PeftModel.from_pretrained(base_model,"reasonwang/flan-alpaca-lora-base")
16
 
17
+ inputs = tokenizer("List a few tips to get good scores in math.", return_tensors="pt")
18
  outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)
19
  print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
20
  ```