diff --git a/setup.py b/setup.py index 17c76bcfb9..835aaabd72 100644 --- a/setup.py +++ b/setup.py @@ -60,8 +60,8 @@ EXTRAS_REQUIRE = { "neural-compressor": ["neural-compressor>=2.2.0", "onnxruntime<1.15.0", "accelerate"], - "openvino": ["openvino-nightly==2024.2.0.dev20240523", "nncf==2.10.0", "openvino-tokenizers[transformers]"], - "nncf": ["nncf==2.10.0"], + "openvino": ["openvino==2024.2.0", "nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v2120", "openvino-tokenizers[transformers]"], + "nncf": ["nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v2120"], "ipex": ["intel-extension-for-pytorch", "transformers>=4.36.0,<4.39.0"], "diffusers": ["diffusers"], "quality": QUALITY_REQUIRE,