replace llava test model

huggingface · Nov 7, 2024 · a3c6506 · a3c6506
1 parent bb90a76
commit a3c6506
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -2007,10 +2007,6 @@ def test_llava_with_new_preprocessing(self, model_arch):
             model_id, export=True, trust_remote_code=model_arch in self.REMOTE_CODE_MODELS
         )
         self.assertTrue(ov_model._support_new_processing)
-        if model_arch == "llava":
-            # testing model for llava does ot have specified image_seq_length and it is different from default
-            transformers_model.config.image_seq_length = 225
-            ov_model.config.image_seq_length = 225
         self.assertTrue(processor.patch_size is not None)
         self.assertTrue(processor.vision_feature_select_strategy is not None)
         inputs = processor(images=self.IMAGE, text=prompt, return_tensors="pt")

diff --git a/tests/openvino/utils_tests.py b/tests/openvino/utils_tests.py
@@ -76,7 +76,7 @@
     "llama": "HuggingFaceM4/tiny-random-LlamaForCausalLM",
     "llama_awq": "HuggingFaceH4/tiny-random-LlamaForCausalLM",
     "llama_gptq": "hf-internal-testing/TinyLlama-1.1B-Chat-v0.3-GPTQ",
-    "llava": "trl-internal-testing/tiny-random-LlavaForConditionalGeneration",
+    "llava": "katuni4ka/tiny-random-llava",
     "llava_next": "katuni4ka/tiny-random-llava-next",
     "m2m_100": "hf-internal-testing/tiny-random-m2m_100",
     "opt": "hf-internal-testing/tiny-random-OPTModel",