add model f679f38
Canwen Xu commited on
How to use canwenxu/ssr-base with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("canwenxu/ssr-base")
model = AutoModelForSeq2SeqLM.from_pretrained("canwenxu/ssr-base", device_map="auto")