Model-Speed-Comparator / download_model.py
Mridul2004's picture
Switch to bert-tiny - fits in 512MB RAM
961a8e6
Raw
History Blame Contribute Delete
279 Bytes
from transformers import AutoTokenizer, AutoModelForSequenceClassification
MODEL = "prajjwal1/bert-tiny"
print("Downloading tokenizer...")
AutoTokenizer.from_pretrained(MODEL)
print("Downloading model...")
AutoModelForSequenceClassification.from_pretrained(MODEL)
print("Done!")