legacy-datasets/mc4
Updated โข 2.58k โข 163
How to use kkuramitsu/mt5-kogi-regio with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("kkuramitsu/mt5-kogi-regio")
model = AutoModelForSeq2SeqLM.from_pretrained("kkuramitsu/mt5-kogi-regio", device_map="auto")