modified: config.json 9c3b31e
muyeby commited on
How to use xfbai/AMRBART-base with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("xfbai/AMRBART-base")
model = AutoModelForSeq2SeqLM.from_pretrained("xfbai/AMRBART-base", device_map="auto")