From 8ba89b469d268c69a1d18d79b4fe8eb9a7731e0d Mon Sep 17 00:00:00 2001 From: Nikita Savelyev Date: Thu, 28 Nov 2024 17:49:00 +0100 Subject: [PATCH] Update test reference for older transformers version --- tests/openvino/test_quantization.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py index cc8ba7fd8..8aad68b23 100644 --- 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), ), ]