Skip to content

Commit cf3aad4

Browse files
Fix qwen2-vl tests (#1084)
1 parent 93777ec commit cf3aad4

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/openvino/test_exporters_cli.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -230,9 +230,9 @@ class OVCLIExportTestCase(unittest.TestCase):
230230
(
231231
"image-text-to-text",
232232
"qwen2_vl",
233-
'int4 --group-size 4 --ratio 0.9 --sensitivity-metric "mean_activation_magnitude" '
233+
'int4 --group-size 16 --ratio 0.8 --sensitivity-metric "mean_activation_magnitude" '
234234
"--dataset contextual --num-samples 1",
235-
{"int8": 8, "int4": 22},
235+
[{"int8": 10, "int4": 20}, {"int8": 1}, {"int8": 1}, {"int8": 10}],
236236
),
237237
]
238238
)

tests/openvino/test_quantization.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -500,7 +500,7 @@ class OVWeightCompressionTest(unittest.TestCase):
500500
sensitivity_metric="mean_activation_magnitude",
501501
num_samples=1,
502502
),
503-
{"int4": 20, "int8": 10},
503+
[{"int8": 10, "int4": 20}, {"int8": 1}, {"int8": 1}, {"int8": 10}],
504504
),
505505
]
506506
)

0 commit comments

Comments
 (0)