Views
No views yet
1import transformers
2from peft import PeftModel
3
4base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
5peft_model = PeftModel.from_pretrained("zirui3/flan-t5-large-alpaca")
6
7inputs = tokenizer("Any instruction that you like.", return_tensors="pt")
8outputs = peft_model.generate(**inputs, max_length=128, do_sample=True)
9print(tokenizer.batch_decode(outputs, skip_special_tokens=True)