Update README.md

README.md

@@ -11,8 +11,10 @@ Usage:
 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")
+model_name = "google/flan-t5-large"; peft_model_id = "reasonwang/flan-alpaca-lora-large"
+tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
+base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained(model_name)
+peft_model = PeftModel.from_pretrained(base_model, peft_model_id)
 
 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)