Update tests/openvino/test_quantization.py

huggingface · Dec 19, 2024 · 7700019 · 7700019
1 parent 6b9dc88
commit 7700019
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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),
         ),
     ]