File size: 431 Bytes
347a08e | 1 2 3 4 5 6 7 8 9 10 11 | # prepare_model.py
from transformers import AutoModelForSequenceClassification, AutoTokenizer
MODEL = "distilbert-base-uncased-finetuned-sst-2-english"
m = AutoModelForSequenceClassification.from_pretrained(MODEL)
t = AutoTokenizer.from_pretrained(MODEL)
m.save_pretrained(".") # saves config.json + weights (bin/safetensors)
t.save_pretrained(".") # saves tokenizer files
print("Saved model + tokenizer to repo root") |