violetch24 commited on
Commit
bab0264
1 Parent(s): d56926a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -32,8 +32,8 @@ The linear modules **roberta.encoder.layer.7.output.dense**, **roberta.encoder.l
32
  ### Load with optimum:
33
 
34
  ```python
35
- from optimum.intel.neural_compressor.quantization import IncQuantizedModelForCausalLM
36
- int8_model = IncQuantizedModelForCausalLM.from_pretrained(
37
  'Intel/roberta-base-squad2-int8-static',
38
  )
39
  ```
 
32
  ### Load with optimum:
33
 
34
  ```python
35
+ from optimum.intel.neural_compressor.quantization import IncQuantizedModelForQuestionAnswering
36
+ int8_model = IncQuantizedModelForQuestionAnswering.from_pretrained(
37
  'Intel/roberta-base-squad2-int8-static',
38
  )
39
  ```