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")