echarlaix HF staff commited on
Commit
7050066
1 Parent(s): a6c0703

update loading instructions

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -52,8 +52,8 @@ Below linear modules (40/193) are fallbacked to fp32 for less than 1% relative a
52
  ### Load with optimum:
53
 
54
  ```python
55
- from optimum.intel.neural_compressor.quantization import IncQuantizedModelForSeq2SeqLM
56
- int8_model = IncQuantizedModelForSeq2SeqLM.from_pretrained(
57
- 'Intel/bart-large-cnn-int8-dynamic',
58
- )
59
- ```
 
52
  ### Load with optimum:
53
 
54
  ```python
55
+ from optimum.intel import INCModelForSeq2SeqLM
56
+
57
+ model_id = "Intel/bart-large-cnn-int8-dynamic"
58
+ int8_model = INCModelForSeq2SeqLM.from_pretrained(model_id)
59
+ ```