Skip to content

Commit 8635b5c

Browse files
authored
Trigger tests for maira2 for compatible transformers version only (#1161)
1 parent 0ff2bbe commit 8635b5c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/openvino/test_modeling.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -2114,7 +2114,10 @@ class OVModelForVisualCausalLMIntegrationTest(unittest.TestCase):
21142114
if is_transformers_version(">=", "4.40.0"):
21152115
SUPPORTED_ARCHITECTURES += ["llava_next", "nanollava"]
21162116
if is_transformers_version(">=", "4.45.0"):
2117-
SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v", "qwen2_vl", "maira2"]
2117+
SUPPORTED_ARCHITECTURES += ["minicpmv", "internvl2", "phi3_v", "qwen2_vl"]
2118+
if is_transformers_version(">=", "4.46.0"):
2119+
SUPPORTED_ARCHITECTURES += ["maira2"]
2120+
21182121
TASK = "image-text-to-text"
21192122
REMOTE_CODE_MODELS = ["internvl2", "minicpmv", "nanollava", "phi3_v", "maira2"]
21202123

0 commit comments

Comments
 (0)