From f6b655a1c7565a87768c48529ece207c21ae4398 Mon Sep 17 00:00:00 2001 From: Austin Liu Date: Wed, 11 Dec 2024 10:19:26 +0800 Subject: [PATCH] Revert workaround of disabling QWEN2_VL convergence tests Signed-off-by: Austin Liu --- test/convergence/test_mini_models_multimodal.py | 11 ++++------- test/convergence/test_mini_models_with_logits.py | 11 ++++------- 2 files changed, 8 insertions(+), 14 deletions(-) diff --git a/test/convergence/test_mini_models_multimodal.py b/test/convergence/test_mini_models_multimodal.py index f67e96c50..13ef7d8c3 100644 --- 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 index 5ca3e7420..68d9f890b 100644 --- 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", ), ], ),