Fix test

nikita-savelyevv · nikita-savelyevv · commit 24c5133b6018 · 2024-10-16T10:08:46.000+02:00
diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
@@ -533,7 +533,7 @@ def test_ovmodel_4bit_auto_compression_with_config(
                 # TODO: Check that AWQ was actually applied
                 pass
 
-            ov_model = model
+            ov_model = model.model
             if model_cls == OVModelForVisualCausalLM:
                 ov_model = model.lm_model
 
