diff --git a/optimum/intel/openvino/configuration.py b/optimum/intel/openvino/configuration.py index 739b5640a6..398e1e0b50 100644 --- a/optimum/intel/openvino/configuration.py +++ b/optimum/intel/openvino/configuration.py @@ -310,7 +310,9 @@ def __init__( if isinstance(quantization_config, dict): quantization_config = self._quantization_config_from_dict(quantization_config) self.quantization_config = quantization_config - self.compression = None # A field for backward-compatability of training-time compression parameters + self.compression = kwargs.get( + "compression", None + ) # A field for backward-compatability of training-time compression parameters bits = self.quantization_config.bits if self.quantization_config else None self.dtype = "int" + str(bits) if isinstance(bits, int) else dtype