armanc/scientific_papers
Updated โข 2.73k โข 186
How to use jaimik69/t5-small-pubmed with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("jaimik69/t5-small-pubmed")
model = AutoModelForSeq2SeqLM.from_pretrained("jaimik69/t5-small-pubmed", device_map="auto")This is a T5-Small Model trained of pubmed dataset of articles and abstracts.