diff --git a/tests/compile/test_dynamic_shapes_compilation.py b/tests/compile/test_dynamic_shapes_compilation.py index b63a4607c..bbd62237c 100644 --- a/tests/compile/test_dynamic_shapes_compilation.py +++ b/tests/compile/test_dynamic_shapes_compilation.py @@ -23,8 +23,14 @@ from vllm.utils.torch_utils import is_torch_equal_or_newer def get_test_models(): """Get list of models to test based on PyTorch version""" - # TODO "Qwen/Qwen3-4B-Instruct-2507" fails Fix issue and support it. - return ["gpt2", "Qwen/Qwen2-7B-Instruct", "meta-llama/Llama-3.1-8B"] + models = [ + "gpt2", + "Qwen/Qwen2-7B-Instruct", + "meta-llama/Llama-3.1-8B", + ] + if is_torch_equal_or_newer("2.12.0"): + models.append("Qwen/Qwen3-4B-Instruct-2507") + return models @pytest.mark.parametrize("model_name", get_test_models())