Safetensors
distilbert
ll24 / load_model.py
chaosbringerc's picture
Create load_model.py
d423c81 verified
Raw
History Blame Contribute Delete
408 Bytes
from transformers import AutoModelForSequenceClassification, AutoTokenizer
def load_ll24():
model = AutoModelForSequenceClassification.from_pretrained(
"ll24",
from_safetensors=True
)
tokenizer = AutoTokenizer.from_pretrained("ll24")
return model, tokenizer
if __name__ == "__main__":
model, tokenizer = load_ll24()
print("Model and tokenizer loaded successfully!")