Revert workaround of disabling QWEN2_VL convergence tests

Signed-off-by: Austin Liu <[email protected]>
linkedin · Dec 11, 2024 · f6b655a · f6b655a
1 parent 78e8a85
commit f6b655a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 14 deletions.
diff --git a/test/convergence/test_mini_models_multimodal.py b/test/convergence/test_mini_models_multimodal.py
@@ -380,9 +380,8 @@ def run_mini_model_multimodal(
             5e-3,
             1e-5,
             marks=pytest.mark.skipif(
-                not QWEN2_VL_AVAILABLE
-                or version.parse(transformers.__version__) >= version.parse("4.47.0"),
-                reason="Qwen2-VL not available in this version of transformers or transformers version >= 4.47.0",
+                not QWEN2_VL_AVAILABLE,
+                reason="Qwen2-VL not available in this version of transformers",
             ),
         ),
         pytest.param(
@@ -401,10 +400,8 @@ def run_mini_model_multimodal(
                     not supports_bfloat16(), reason="bfloat16 not supported on this GPU"
                 ),
                 pytest.mark.skipif(
-                    not QWEN2_VL_AVAILABLE
-                    or version.parse(transformers.__version__)
-                    >= version.parse("4.47.0"),
-                    reason="Qwen2-VL not available in this version of transformers or transformers version >= 4.47.0",
+                    not QWEN2_VL_AVAILABLE,
+                    reason="Qwen2-VL not available in this version of transformers",
                 ),
             ],
         ),

diff --git a/test/convergence/test_mini_models_with_logits.py b/test/convergence/test_mini_models_with_logits.py
@@ -540,9 +540,8 @@ def run_mini_model(
             5e-3,
             1e-5,
             marks=pytest.mark.skipif(
-                not QWEN2_VL_AVAILABLE
-                or version.parse(transformers.__version__) >= version.parse("4.47.0"),
-                reason="Qwen2-VL not available in this version of transformers or transformers version >= 4.47.0",
+                not QWEN2_VL_AVAILABLE,
+                reason="Qwen2-VL not available in this version of transformers",
             ),
         ),
         pytest.param(
@@ -561,10 +560,8 @@ def run_mini_model(
                     not supports_bfloat16(), reason="bfloat16 not supported on this GPU"
                 ),
                 pytest.mark.skipif(
-                    not QWEN2_VL_AVAILABLE
-                    or version.parse(transformers.__version__)
-                    >= version.parse("4.47.0"),
-                    reason="Qwen2-VL not available in this version of transformers or transformers version >= 4.47.0",
+                    not QWEN2_VL_AVAILABLE,
+                    reason="Qwen2-VL not available in this version of transformers",
                 ),
             ],
         ),