echarlaix HF staff commited on
Commit
085de21
1 Parent(s): e9ab251

update loading instructions

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -34,8 +34,8 @@ The original fp32 model comes from the fine-tuned model [gchhablani/bert-base-ca
34
  #### Load with optimum:
35
 
36
  ```python
37
- from optimum.intel.neural_compressor.quantization import IncQuantizedModelForSequenceClassification
38
- int8_model = IncQuantizedModelForSequenceClassification.from_pretrained(
39
- 'Intel/bert-base-cased-finetuned-sst2-int8',
40
- )
41
- ```
 
34
  #### Load with optimum:
35
 
36
  ```python
37
+ from optimum.intel import INCModelForSequenceClassification
38
+
39
+ model_id = "Intel/bert-base-cased-finetuned-sst2-int8"
40
+ int8_model = INCModelForSequenceClassification.from_pretrained(model_id)
41
+ ```