Unverified Commit 112944fa authored by Laith Sakka's avatar Laith Sakka Committed by GitHub
Browse files

test Qwen/Qwen3-4B-Instruct-2507 for unbacked (#36064)


Signed-off-by: default avatarLaith Sakka <lsakka@meta.com>
parent 91be5f9b
...@@ -23,8 +23,14 @@ from vllm.utils.torch_utils import is_torch_equal_or_newer ...@@ -23,8 +23,14 @@ from vllm.utils.torch_utils import is_torch_equal_or_newer
def get_test_models(): def get_test_models():
"""Get list of models to test based on PyTorch version""" """Get list of models to test based on PyTorch version"""
# TODO "Qwen/Qwen3-4B-Instruct-2507" fails Fix issue and support it. models = [
return ["gpt2", "Qwen/Qwen2-7B-Instruct", "meta-llama/Llama-3.1-8B"] "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()) @pytest.mark.parametrize("model_name", get_test_models())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment