Skip to content

Commit f928019

Browse files
Check PR for torch 2.5 supportd
1 parent d357376 commit f928019

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

setup.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,12 @@
6161
QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]
6262

6363
EXTRAS_REQUIRE = {
64-
"nncf": ["nncf>=2.11.0"],
65-
"openvino": ["nncf>=2.11.0", "openvino==2024.4.1.dev20240926", "openvino-tokenizers==2024.4.1.0.dev20240926"],
64+
"nncf": ["nncf@git+https://github.com/anzr299/nncf.git@torch250"],
65+
"openvino": [
66+
"nncf@git+https://github.com/anzr299/nncf.git@torch250",
67+
"openvino==2024.4.1.dev20240926",
68+
"openvino-tokenizers==2024.4.1.0.dev20240926"
69+
],
6670
"neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"],
6771
"ipex": ["intel-extension-for-pytorch", "transformers>=4.39,<4.45"],
6872
"diffusers": ["diffusers"],

0 commit comments

Comments
 (0)