From 35a30a011e295b7b6c7779e2ce679f91e00e4946 Mon Sep 17 00:00:00 2001 From: Alexander Date: Tue, 14 Nov 2023 09:57:51 +0400 Subject: [PATCH] Fixed test --- tests/openvino/test_quantization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py index de464d10a0..6884cba081 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, 104, 84), + ("hf-internal-testing/tiny-random-t5", 30, 32, 52, 84), ) @parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS)