reasonwang
/

flan-alpaca-lora-base

Model card Files Files and versions Community

reasonwang commited on Mar 29, 2023

Commit

d9348be

·

1 Parent(s): 2dd5eef

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -11,8 +11,8 @@ Usage:
 import transformers
 from peft import PeftModel
-base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
-peft_model = PeftModel.from_pretrained(base_model,"reasonwang/flan-alpaca-lora-base")
 inputs = tokenizer("List a few tips to get good scores in math.", return_tensors="pt")
 outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)

 import transformers
 from peft import PeftModel
+base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
+peft_model = PeftModel.from_pretrained(base_model,"reasonwang/flan-alpaca-lora-large")
 inputs = tokenizer("List a few tips to get good scores in math.", return_tensors="pt")
 outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)