diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py index fa94839a6..8d4780e29 100644 --- a/tests/openvino/test_quantization.py +++ b/tests/openvino/test_quantization.py @@ -106,7 +106,7 @@ class OVQuantizerTest(unittest.TestCase): weight_only=False, smooth_quant_alpha=0.95, ), - (14, 22, 21) if is_transformers_version("<=", "4.42.4") else (14, 26, 25), + (14, 22, 21) if is_transformers_version("<=", "4.42.4") else (14, 22, 25), (14, 21, 17) if is_transformers_version("<=", "4.42.4") else (14, 22, 18), ), ]