rexarski/eli5_category
Updated • 517 • 20
How to use jsgao/bart-eli5c with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("jsgao/bart-eli5c")
model = AutoModelForSeq2SeqLM.from_pretrained("jsgao/bart-eli5c", device_map="auto")Answer generator model of ELI5-Category Dataset