From f92801962b507a19e3bdf9f49ca559594d4a6dfb Mon Sep 17 00:00:00 2001
From: Nikita Savelyev <nikita.savelyev@intel.com>
Date: Wed, 6 Nov 2024 10:07:53 +0100
Subject: [PATCH] Check PR for torch 2.5 supportd

---
 setup.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 7ef3652f87..b044985db7 100644
--- 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"],