Fix qwen2-vl tests

huggingface · Dec 19, 2024 · 6a40e44 · 6a40e44
1 parent 93777ec
commit 6a40e44
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/openvino/test_exporters_cli.py b/tests/openvino/test_exporters_cli.py
@@ -230,9 +230,9 @@ class OVCLIExportTestCase(unittest.TestCase):
                 (
                     "image-text-to-text",
                     "qwen2_vl",
-                    'int4 --group-size 4 --ratio 0.9 --sensitivity-metric "mean_activation_magnitude" '
+                    'int4 --group-size 16 --ratio 0.8 --sensitivity-metric "mean_activation_magnitude" '
                     "--dataset contextual --num-samples 1",
-                    {"int8": 8, "int4": 22},
+                    [{"int8": 10, "int4": 20}, {"int8": 1}, {"int8": 1}, {"int8": 10}],
                 ),
             ]
         )

diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
@@ -500,7 +500,7 @@ class OVWeightCompressionTest(unittest.TestCase):
                         sensitivity_metric="mean_activation_magnitude",
                         num_samples=1,
                     ),
-                    {"int4": 20, "int8": 10},
+                    [{"int8": 10, "int4": 20}, {"int8": 1}, {"int8": 1}, {"int8": 10}],
                 ),
             ]
         )