diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py index 6884cba081..00fa36b21b 100644 --- a/tests/openvino/test_quantization.py +++ b/tests/openvino/test_quantization.py @@ -146,7 +146,7 @@ def preprocess_function(examples, tokenizer): class OVQuantizerSeq2SeqTest(unittest.TestCase): SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS = ( - ("hf-internal-testing/tiny-random-t5", 30, 32, 52, 84), + ("hf-internal-testing/tiny-random-t5", 30, 32, 52, 42), ) @parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS)