How to use nlpunibo/distilbert_classifier2 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, DistilBertClassifier tokenizer = AutoTokenizer.from_pretrained("nlpunibo/distilbert_classifier2") model = DistilBertClassifier.from_pretrained("nlpunibo/distilbert_classifier2", device_map="auto")
SFconvertbot