Skip to content

Commit 429e34c

Browse files
committed
add accelerate dependency for inc extra
1 parent ebc108b commit 429e34c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@
3939
QUALITY_REQUIRE = ["black~=23.1", "ruff>=0.0.241"]
4040

4141
EXTRAS_REQUIRE = {
42-
"neural-compressor": ["neural-compressor>=2.2.0", "onnx", "onnxruntime<1.15.0"],
42+
"neural-compressor": ["neural-compressor>=2.2.0", "onnx", "onnxruntime<1.15.0", "accelerate"],
4343
"openvino": ["openvino>=2023.3", "onnx", "nncf>=2.8.1"],
4444
"openvino-tokenizers": ["openvino-tokenizers[transformers]"],
4545
"nncf": ["nncf>=2.8.1"],

0 commit comments

Comments
 (0)