Views
No views yet
1import transformers
2from peft import PeftModel
3
4model_name = "google/flan-t5-large"; peft_model_id = "reasonwang/flan-alpaca-lora-large"
5tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
6base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained(model_name)
7peft_model = PeftModel.from_pretrained(base_model, peft_model_id)
8
9inputs = tokenizer("List a few tips to get good scores in math.", return_tensors="pt")
10outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)
11print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
12