Skip to content

Commit 9c24871

Browse files
committed
Limit intel-extension-for-transformers version
Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
1 parent af07192 commit 9c24871

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

optimum/intel/neural_compressor/quantization.py

+9
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,11 @@
4747

4848
from ..utils.constant import _TASK_ALIASES, MIN_QDQ_ONNX_OPSET, ONNX_WEIGHTS_NAME, WEIGHTS_NAME
4949
from ..utils.import_utils import (
50+
_intel_extension_for_transformers_version,
5051
_ipex_version,
5152
_neural_compressor_version,
5253
is_intel_extension_for_transformers_available,
54+
is_intel_extension_for_transformers_version,
5355
is_ipex_version,
5456
is_neural_compressor_version,
5557
)
@@ -73,6 +75,13 @@
7375
from intel_extension_for_transformers.transformers.utils.config import WeightOnlyQuantConfig
7476

7577
Config = Union[PostTrainingQuantConfig, WeightOnlyQuantConfig]
78+
79+
INTEL_EXTENSION_FOR_TRANSFORMERS_MINIMUM_VERSION = "1.3.2"
80+
if is_intel_extension_for_transformers_version("!=", INTEL_EXTENSION_FOR_TRANSFORMERS_MINIMUM_VERSION):
81+
raise ImportError(
82+
f"Found an incompatible version of `intel-extension-for-transformers`. Found version {_intel_extension_for_transformers_version}, "
83+
f"but only version {INTEL_EXTENSION_FOR_TRANSFORMERS_MINIMUM_VERSION} is supported."
84+
)
7685
else:
7786
Config = PostTrainingQuantConfig
7887

0 commit comments

Comments
 (0)