nyu-mll/glue
Viewer • Updated • 1.49M • 472k • 504
How to use Intel/distilbert-base-uncased-MRPC-int8-dynamic-inc with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Intel/distilbert-base-uncased-MRPC-int8-dynamic-inc") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Intel/distilbert-base-uncased-MRPC-int8-dynamic-inc")
model = AutoModelForSequenceClassification.from_pretrained("Intel/distilbert-base-uncased-MRPC-int8-dynamic-inc")Model Description: This model is a DistilBERT fine-tuned on MPRC dynamically quantized with optimum-intel through the usage of huggingface/optimum-intel through the usage of Intel® Neural Compressor.
To load the quantized model, you can do as follows:
from optimum.intel import INCModelForSequenceClassification
model_id = "Intel/distilbert-base-uncased-MRPC-int8-dynamic"
model = INCModelForSequenceClassification.from_pretrained(model_id)
| INT8 | FP32 | |
|---|---|---|
| Accuracy (eval-f1) | 0.8983 | 0.9027 |
| Model size (MB) | 75 | 268 |