From bd777cbe4e6b0fa227692569100eda74e1a80572 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Wed, 29 May 2024 11:13:52 +0200 Subject: [PATCH] fix expected ops quant in test --- tests/openvino/test_quantization.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/openvino/test_quantization.py b/tests/openvino/test_quantization.py index 6cb6171fa..b7ed36d3e 100644 --- a/tests/openvino/test_quantization.py +++ b/tests/openvino/test_quantization.py @@ -79,8 +79,7 @@ class OVQuantizerTest(unittest.TestCase): ) SUPPORTED_ARCHITECTURES_OV_MODEL = ( (OVModelForSequenceClassification, "bert", 32, 35), - (OVModelForCausalLM, "gpt2", 31, 3), - # (OVModelForCausalLM, "gpt2", 31, 22), + (OVModelForCausalLM, "gpt2", 31, 22), ) @parameterized.expand(SUPPORTED_ARCHITECTURES_TORCH_MODEL)