Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2from peft import PeftModel
3
4base = AutoModelForSeq2SeqLM.from_pretrained("sshleifer/distilbart-cnn-12-6")
5tokenizer = AutoTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
6
7model = PeftModel.from_pretrained(base, "username/distilbart-samsum-lora")