Apply suggestions from code review

Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com>
huggingface · Oct 11, 2024 · 1b1b4fe · 1b1b4fe
1 parent bddbdd2
commit 1b1b4fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: