From 1b1b4fe80f186eae6031b5ac4c81a6ca40941f13 Mon Sep 17 00:00:00 2001 From: Ekaterina Aidova Date: Fri, 11 Oct 2024 15:24:33 +0400 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com> --- tests/openvino/test_modeling.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py index ac2145394..b53f3a7b2 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: