echarlaix HF staff commited on
Commit
210504a
1 Parent(s): b7c4022

update loading instructions

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -51,10 +51,10 @@ The calibration dataloader is the train dataloader. The default calibration samp
51
  #### Load with optimum:
52
 
53
  ```python
54
- from optimum.intel.neural_compressor.quantization import IncQuantizedModelForSequenceClassification
55
- int8_model = IncQuantizedModelForSequenceClassification.from_pretrained(
56
- 'Intel/electra-small-discriminator-mrpc-int8-static',
57
- )
58
  ```
59
 
60
  ### ONNX
 
51
  #### Load with optimum:
52
 
53
  ```python
54
+ from optimum.intel import INCModelForSequenceClassification
55
+
56
+ model_id = "Intel/electra-small-discriminator-mrpc-int8-static"
57
+ int8_model = INCModelForSequenceClassification.from_pretrained(model_id)
58
  ```
59
 
60
  ### ONNX