Skip to content

Commit 5d2882b

Browse files
Tweak test
1 parent 2ad2730 commit 5d2882b

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/openvino/test_quantization.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -103,10 +103,9 @@ class OVQuantizerTest(unittest.TestCase):
103103
processor=MODEL_NAMES["whisper"],
104104
trust_remote_code=True,
105105
weight_only=False,
106-
smooth_quant_alpha=0.95,
107106
),
108-
(14, 22, 25),
109-
(14, 22, 18),
107+
(14, 21 if is_transformers_version("<=", "4.36.0") else 22, 25),
108+
(14, 21 if is_transformers_version("<=", "4.36.0") else 22, 18),
110109
),
111110
]
112111

0 commit comments

Comments
 (0)