fix tests

Author: eaidova

tests/openvino/test_exporters_cli.py

@@ -269,7 +269,7 @@ def test_exporters_cli_int8(self, task: str, model_type: str):
 
             if task.startswith("text2text-generation"):
                 models = [model.encoder, model.decoder]
-                if task.endswith("with-past"):
+                if task.endswith("with-past") and not model.decoder.stateful:
                     models.append(model.decoder_with_past)
             elif model_type.startswith("stable-diffusion") or model_type.startswith("flux"):
                 models = [model.unet or model.transformer, model.vae_encoder, model.vae_decoder]

tests/openvino/test_modeling.py

@@ -1527,8 +1527,9 @@ def test_compare_to_transformers(self, model_arch):
 
         self.assertIsInstance(ov_model.encoder, OVEncoder)
         self.assertIsInstance(ov_model.decoder, OVDecoder)
-        self.assertIsInstance(ov_model.decoder_with_past, OVDecoder)
-        self.assertIsInstance(ov_model.config, PretrainedConfig)
+        if not ov_model.decoder.stateful:
+            self.assertIsInstance(ov_model.decoder_with_past, OVDecoder)
+            self.assertIsInstance(ov_model.config, PretrainedConfig)
 
         transformers_model = AutoModelForSeq2SeqLM.from_pretrained(model_id)
         tokenizer = AutoTokenizer.from_pretrained(model_id)
@@ -1613,7 +1614,7 @@ def test_generate_utils(self, model_arch):
         gc.collect()
 
     def test_compare_with_and_without_past_key_values(self):
-        model_id = MODEL_NAMES["t5"]
+        model_id = MODEL_NAMES["bart"]
         tokenizer = AutoTokenizer.from_pretrained(model_id)
         text = "This is a sample input"
         tokens = tokenizer(text, return_tensors="pt")

tests/openvino/test_quantization.py

@@ -106,7 +106,7 @@ class OVQuantizerTest(unittest.TestCase):
                 weight_only=False,
                 smooth_quant_alpha=0.95,
             ),
-            (14, 22, 21) if is_transformers_version("<=", "4.42.4") else (14, 22, 25),
+            (14, 22, 21) if is_transformers_version("<=", "4.42.4") else (14, 26, 25),
             (14, 21, 17) if is_transformers_version("<=", "4.42.4") else (14, 22, 18),
         ),
     ]
@@ -1213,7 +1213,7 @@ def test_calibration_data_uniqueness(self, model_name, apply_caching):
         processor = AutoProcessor.from_pretrained(model_id)
 
         calibration_data = []
-        if not ov_model.stateful:
+        if not ov_model.decoder.stateful:
             ov_model.decoder_with_past.request = InferRequestWrapper(
                 ov_model.decoder_with_past.request, calibration_data, apply_caching=apply_caching
             )