Training in progress, epoch 1 82f16f2
Eliott Z commited on
How to use eliolio/bart-finetuned-yelpreviews with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("eliolio/bart-finetuned-yelpreviews")
model = AutoModelForSeq2SeqLM.from_pretrained("eliolio/bart-finetuned-yelpreviews", device_map="auto")