Update test reference for older transformers version

huggingface · Nov 28, 2024 · 8ba89b4 · 8ba89b4
1 parent ddc8328
commit 8ba89b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py
@@ -104,8 +104,8 @@ class OVQuantizerTest(unittest.TestCase):
                 trust_remote_code=True,
                 weight_only=False,
             ),
-            (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),
         ),
     ]