Fixed test

huggingface · Feb 7, 2024 · ae1da0f · ae1da0f
1 parent 4707914
commit ae1da0f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
@@ -286,10 +286,15 @@ def test_ovmodel_4bit_weight_compression(self, model_cls, model_name, expected_i
                 tokenizer.pad_token = tokenizer.eos_token
 
             quantizer = OVQuantizer.from_pretrained(transformers_model, task=task)
+            ov_config = OVConfig(
+                weight_quantization_config=OVWeightQuantizationConfig(
+                    mode=nncf.CompressWeightsMode.INT4_SYM, ratio=0.8
+                )
+            )
             quantizer.quantize(
                 save_directory=tmp_dir,
                 weights_only=True,
-                quantization_config=OVWeightQuantizationConfig(mode=nncf.CompressWeightsMode.INT4_SYM, ratio=0.8),
+                ov_config=ov_config,
             )
             model = model_cls.from_pretrained(tmp_dir)