From 846f1e7a33258e3755874aee06d34ecfb9f47f1e Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Thu, 13 Feb 2025 17:18:09 +0100 Subject: [PATCH] Trigger tests for maira2 for compatible transformers version --- tests/openvino/test_modeling.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py index 3a1ba74dc9..60b7576973 100644 --- a/tests/openvino/test_modeling.py +++ b/tests/openvino/test_modeling.py @@ -2114,7 +2114,10 @@ class OVModelForVisualCausalLMIntegrationTest(unittest.TestCase): if is_transformers_version(">=", "4.40.0"): SUPPORTED_ARCHITECTURES += ["llava_next", "nanollava"] if is_transformers_version(">=", "4.45.0"): - SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v", "qwen2_vl", "maira2"] + SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v", "qwen2_vl"] + if is_transformers_version(">=", "4.46.0"): + SUPPORTED_ARCHITECTURES += ["maira2"] + TASK = "image-text-to-text" REMOTE_CODE_MODELS = ["internvl2", "minicpmv", "nanollava", "phi3_v", "maira2"]