fix

Author: echarlaix

optimum/intel/openvino/quantization.py

@@ -234,7 +234,7 @@ def quantize(
         if ov_config is not None:
             if not isinstance(ov_config, OVConfig):
                 raise TypeError(f"`ov_config` should be an `OVConfig`, but got: {type(ov_config)} instead.")
-            elif not ov_config.compression:
+            elif ov_config.compression is None:
                 ov_config.compression = DEFAULT_QUANTIZATION_CONFIG
 
         if isinstance(self.model, OVBaseModel):

optimum/intel/openvino/trainer.py

@@ -226,7 +226,7 @@ def __init__(
         self.compression_controller = None
 
         if self.ov_config is not None:
-            if not self.ov_config.compression:
+            if self.ov_config.compression is None:
                 self.ov_config.compression = DEFAULT_QUANTIZATION_CONFIG
 
             if self.args.do_train:

tests/openvino/test_quantization.py

@@ -255,7 +255,7 @@ def test_automodel_weight_compression(self, model_cls, model_name, expected_pt_i
             self.assertTrue("logits" in outputs)
 
             # Verify that that the configuration is correctly saved and loaded
-            openvino_config = OVConfig.from_pretrained(tmp_dir)
+            loaded_config = OVConfig.from_pretrained(tmp_dir)
             self.assertIsNotNone(loaded_config)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_EXPECTED_8BIT_COMPRESSED_MATMULS)