Tweak test

huggingface · Dec 3, 2024 · 5d2882b · 5d2882b
1 parent 2ad2730
commit 5d2882b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
@@ -103,10 +103,9 @@ class OVQuantizerTest(unittest.TestCase):
                 processor=MODEL_NAMES["whisper"],
                 trust_remote_code=True,
                 weight_only=False,
-                smooth_quant_alpha=0.95,
             ),
-            (14, 22, 25),
-            (14, 22, 18),
+            (14, 21 if is_transformers_version("<=", "4.36.0") else 22, 25),
+            (14, 21 if is_transformers_version("<=", "4.36.0") else 22, 18),
         ),
     ]