limit neural compressor version (#871)

huggingface · Aug 19, 2024 · 4c15907 · 4c15907
1 parent 4e1d405
commit 4c15907
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -59,7 +59,7 @@
 QUALITY_REQUIRE = ["black~=23.1", "ruff>=0.0.241"]
 
 EXTRAS_REQUIRE = {
-    "neural-compressor": ["neural-compressor>=2.2.0", "onnxruntime<1.15.0", "accelerate", "transformers<4.43.0"],
+    "neural-compressor": ["neural-compressor>=2.2.0,<3.0", "onnxruntime<1.15.0", "accelerate", "transformers<4.43.0"],
     "openvino": ["openvino>=2023.3", "nncf>=2.11.0", "openvino-tokenizers[transformers]"],
     "nncf": ["nncf>=2.11.0"],
     "ipex": ["intel-extension-for-pytorch", "transformers>=4.39.0,<4.44.0"],