Skip to content

Commit 0991631

Browse files
committed
set neuralcompressor commit
Signed-off-by: changwa1 <chang1.wang@intel.com>
1 parent 18855fd commit 0991631

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

setup.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,11 @@
6363
EXTRAS_REQUIRE = {
6464
"nncf": ["nncf>=2.11.0"],
6565
"openvino": ["nncf>=2.11.0", "openvino>=2024.5.0", "openvino-tokenizers>=2024.5.0"],
66-
"neural-compressor": ["neural-compressor[pt]>3.0", "accelerate", "transformers<4.46"],
66+
"neural-compressor": [
67+
"neural_compressor[pt]@git+https://github.com/intel/neural-compressor.git@5c72158a6799bdf0334ef36fbd493eeed3b62d9f",
68+
"accelerate",
69+
"transformers<4.46",
70+
],
6771
"ipex": ["intel-extension-for-pytorch", "transformers>=4.39,<4.45"],
6872
"diffusers": ["diffusers"],
6973
"quality": QUALITY_REQUIRE,

0 commit comments

Comments
 (0)