Check PR for torch 2.5 supportd

huggingface · Nov 6, 2024 · f928019 · f928019
1 parent d357376
commit f928019
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -61,8 +61,12 @@
 QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
 
 EXTRAS_REQUIRE = {
-    "nncf": ["nncf>=2.11.0"],
-    "openvino": ["nncf>=2.11.0", "openvino==2024.4.1.dev20240926", "openvino-tokenizers==2024.4.1.0.dev20240926"],
+    "nncf": ["nncf@git+https://github.com/anzr299/nncf.git@torch250"],
+    "openvino": [
+        "nncf@git+https://github.com/anzr299/nncf.git@torch250",
+        "openvino==2024.4.1.dev20240926",
+        "openvino-tokenizers==2024.4.1.0.dev20240926"
+    ],
     "neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"],
     "ipex": ["intel-extension-for-pytorch", "transformers>=4.39,<4.45"],
     "diffusers": ["diffusers"],