Add dependency for neural-compressor extras

huggingface · Oct 17, 2023 · cebfcdc · cebfcdc
1 parent 9dd9fec
commit cebfcdc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -41,6 +41,7 @@
         "neural-compressor>=2.2.0",
         "onnx",
         "onnxruntime<1.15.0",
+        "intel-extension-for-transformers",
     ],
     "openvino": ["openvino>=2023.1.0", "onnx", "onnxruntime"],
     "nncf": ["nncf>=2.6.0"],