diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py index ac21453946..b53f3a7b2d 100644 --- a/tests/openvino/test_modeling.py +++ b/tests/openvino/test_modeling.py @@ -1868,7 +1868,7 @@ def test_compare_to_transformers(self, model_arch): ) inputs = processor(images=[self.IMAGE.resize((600, 600))], text=[prompt], return_tensors="pt") ov_model = OVModelForVisualCausalLM.from_pretrained( - model_id, export=True, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS + model_id, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS ) self.assertIsInstance(ov_model, MODEL_TYPE_TO_CLS_MAPPING[ov_model.config.model_type]) self.assertIsInstance(ov_model.vision_embeddings, OVVisionEmbedding) @@ -1914,7 +1914,7 @@ def test_compare_to_transformers(self, model_arch): def test_generate_utils(self, model_arch): model_id = MODEL_NAMES[model_arch] model = OVModelForVisualCausalLM.from_pretrained( - model_id, export=True, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS + model_id, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS ) preprocessor = AutoProcessor.from_pretrained(model_id, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS) if "llava" in model_arch: