Skip to content

Commit 1a821b6

Browse files
authored
Include nncf in openvino extra (#586)
1 parent f1984f4 commit 1a821b6

File tree

4 files changed

+4
-4
lines changed

4 files changed

+4
-4
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ To install the latest release of 🤗 Optimum Intel with the corresponding requi
2020
| Accelerator | Installation |
2121
|:-----------------------------------------------------------------------------------------------------------------|:---------------------------------------------------------------------|
2222
| [Intel Neural Compressor](https://www.intel.com/content/www/us/en/developer/tools/oneapi/neural-compressor.html) | `pip install --upgrade-strategy eager "optimum[neural-compressor]"` |
23-
| [OpenVINO](https://docs.openvino.ai/latest/index.html) | `pip install --upgrade-strategy eager "optimum[openvino,nncf]"` |
23+
| [OpenVINO](https://docs.openvino.ai/latest/index.html) | `pip install --upgrade-strategy eager "optimum[openvino]"` |
2424
| [Intel Extension for PyTorch](https://intel.github.io/intel-extension-for-pytorch/#introduction) | `pip install --upgrade-strategy eager "optimum[ipex]"` |
2525

2626
The `--upgrade-strategy eager` option is needed to ensure `optimum-intel` is upgraded to the latest version.

docs/source/installation.mdx

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ To install the latest release of 🤗 Optimum Intel with the corresponding requi
2121
| Accelerator | Installation |
2222
|:-----------------------------------------------------------------------------------------------------------------------|:-------------------------------------------------------------------|
2323
| [Intel Neural Compressor (INC)](https://www.intel.com/content/www/us/en/developer/tools/oneapi/neural-compressor.html) | `pip install --upgrade-strategy eager "optimum[neural-compressor]"`|
24-
| [Intel OpenVINO](https://docs.openvino.ai/latest/index.html) | `pip install --upgrade-strategy eager "optimum[openvino,nncf]"` |
24+
| [Intel OpenVINO](https://docs.openvino.ai/latest/index.html) | `pip install --upgrade-strategy eager "optimum[openvino]"` |
2525

2626
The `--upgrade-strategy eager` option is needed to ensure `optimum-intel` is upgraded to the latest version.
2727

optimum/exporters/openvino/model_patcher.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def patch_model_with_bettertransformer(model):
3535
+ "[WARNING] For good performance with stateful models, transformers>=4.36.2 and PyTorch>=2.1.1 are required. "
3636
f"This Python environment has Transformers {_transformers_version} and PyTorch {_torch_version}. "
3737
"Consider upgrading PyTorch and Transformers, for example by running "
38-
"`pip install --upgrade --upgrade-strategy eager optimum[openvino,nncf]`, and export the model again"
38+
"`pip install --upgrade --upgrade-strategy eager optimum[openvino]`, and export the model again"
3939
+ COLOR_RESET
4040
)
4141

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040

4141
EXTRAS_REQUIRE = {
4242
"neural-compressor": ["neural-compressor>=2.2.0", "onnx", "onnxruntime<1.15.0"],
43-
"openvino": ["openvino>=2023.3", "onnx", "onnxruntime"],
43+
"openvino": ["openvino>=2023.3", "onnx", "onnxruntime", "nncf>=2.8.1"],
4444
"openvino-tokenizers": ["openvino-tokenizers[transformers]"],
4545
"nncf": ["nncf>=2.8.1"],
4646
"ipex": ["intel-extension-for-pytorch", "onnx"],

0 commit comments

Comments
 (0)