fix expected ops quant in test

huggingface · May 29, 2024 · bd777cb · bd777cb
1 parent 2163bf5
commit bd777cb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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)