Upload 8 files f1001e5 verified
Jedaicoder commited on
How to use mukundagg/bart-large-custom-gec with PEFT:
from peft import PeftModel
from transformers import AutoModelForSeq2SeqLM
base_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large")
model = PeftModel.from_pretrained(base_model, "mukundagg/bart-large-custom-gec")