How to use Ro551/mbart-large-50-GEC-spanish-LORA-merged with PEFT:
from peft import PeftModel from transformers import AutoModelForSeq2SeqLM base_model = AutoModelForSeq2SeqLM.from_pretrained("facebook/mbart-large-50") model = PeftModel.from_pretrained(base_model, "Ro551/mbart-large-50-GEC-spanish-LORA-merged")
How to use Ro551/mbart-large-50-GEC-spanish-LORA-merged with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("Ro551/mbart-large-50-GEC-spanish-LORA-merged", device_map="auto")
How to fix it?
