Models based on Wikipedia
Collection
Uses Wikipedia pages, summaries and categories to build interesting resource-efficient models • 4 items • Updated
How to use tarekziade/wikipedia-summaries-t5-small with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("tarekziade/wikipedia-summaries-t5-small")
model = AutoModelForSeq2SeqLM.from_pretrained("tarekziade/wikipedia-summaries-t5-small", device_map="auto")No model card