Fix

huggingface · Jan 22, 2025 · 602e247 · 602e247
1 parent 43cc1a1
commit 602e247
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -63,8 +63,8 @@
 QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
 
 EXTRAS_REQUIRE = {
-    "nncf": ["nncf@git+https://github.com/nikita-savelyevv/nncf.git@ad/compress-via-openvino"],
-    "openvino": ["nncf@git+https://github.com/nikita-savelyevv/nncf.git@ad/compress-via-openvino", "openvino>=2024.5.0", "openvino-tokenizers>=2024.5.0"],
+    "nncf": ["nncf@git+https://github.com/nikita-savelyevv/nncf.git@compress-via-openvino"],
+    "openvino": ["nncf@git+https://github.com/nikita-savelyevv/nncf.git@compress-via-openvino", "openvino>=2024.5.0", "openvino-tokenizers>=2024.5.0"],
     "neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"],
     "ipex": ["intel-extension-for-pytorch>=2.4", "transformers>4.46,<4.48", "accelerate"],
     "diffusers": ["diffusers"],