comment_summarize / model_downloader.py
aurelien
1st commit
21f5d8a
raw
history blame contribute delete
258 Bytes
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM; \
import os; \
name=os.getenv("MODEL_NAME","facebook/bart-large-cnn"); \
tok=AutoTokenizer.from_pretrained(name); \
_ = AutoModelForSeq2SeqLM.from_pretrained(name); \
print("Model cached:", name)