summit / download_model.py
kkastr
rename files. added prelims for deployment on aws (pending permission fix)
1d197a9
raw
history blame
No virus
289 Bytes
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
tokenizer = AutoTokenizer.from_pretrained("sshleifer/distilbart-cnn-12-6")
model = AutoModelForSeq2SeqLM.from_pretrained("sshleifer/distilbart-cnn-12-6")
tokenizer.save_pretrained("./model")
model.save_pretrained("./model")