DistilBERT base sin mayúsculas ajustado en SST-2 cuantificado dinámicamente

Esto requiere instalar Optimum:

pip install optimum[neural-compressor]

Para cargar el modelo cuantificado y ejecutar la inferencia usando las canalizaciones de Transformers, puedes hacer lo siguiente:

from transformers import AutoTokenizer, pipeline
from optimum.intel import INCModelForSequenceClassification

model_id = "echarlaix/distilbert-base-uncased-finetuned-sst-2-english-int8-dynamic"
model = INCModelForSequenceClassification.from_pretrained(model_id)
tokenizer = AutoTokenizer.from_pretrained(model_id)
cls_pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
text = "He's a dreadful magician."
outputs = cls_pipe(text)