add models 2fddb91
Jining Yu commited on
How to use spicard/small-10 with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("spicard/small-10")
model = AutoModelForSeq2SeqLM.from_pretrained("spicard/small-10", device_map="auto")