NNCF 2.12 & OV 2024.2

huggingface · Jul 26, 2024 · 12814fc · 12814fc
1 parent 59b8a12
commit 12814fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,