fix transformers version

huggingface · Sep 18, 2024 · 60ecb78 · 60ecb78
1 parent cd48153
commit 60ecb78
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -1822,7 +1822,10 @@ def test_compare_with_and_without_past_key_values(self):
 
 
 class OVModelForVisualCausalLMIntegrationTest(unittest.TestCase):
-    SUPPORTED_ARCHITECTURES = ("llava", "llava_next")
+    SUPPORTED_ARCHITECTURES = ["llava", ]
+
+    if is_transformers_version(">=", "4.40.0"):
+        SUPPORTED_ARCHITECTURES += ["llava_next"]
     TASK = "image-text-to-text"
 
     IMAGE = Image.open(