How to use kallacharanteja/mt5-small-to-Tel-medium with PEFT:
from peft import PeftModel from transformers import AutoModelForSeq2SeqLM base_model = AutoModelForSeq2SeqLM.from_pretrained("google/mt5-small") model = PeftModel.from_pretrained(base_model, "kallacharanteja/mt5-small-to-Tel-medium")
How to use kallacharanteja/mt5-small-to-Tel-medium with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("kallacharanteja/mt5-small-to-Tel-medium", device_map="auto")
kallacharanteja commited on 