diff --git a/src/art/megatron/model_support/registry.py b/src/art/megatron/model_support/registry.py index 87f88793e..7dc9c785f 100644 --- a/src/art/megatron/model_support/registry.py +++ b/src/art/megatron/model_support/registry.py @@ -171,6 +171,7 @@ handler_key=_QWEN3_5_DENSE_HANDLER_KEY, model_names=( "Qwen/Qwen3.5-4B", + "Qwen/Qwen3.5-9B", "Qwen/Qwen3.5-27B", "Qwen/Qwen3.6-27B", "Qwen/Qwen3.8-27B", diff --git a/tests/integration/megatron/model_support/test_provider_support.py b/tests/integration/megatron/model_support/test_provider_support.py index c4067cec9..15b056335 100644 --- a/tests/integration/megatron/model_support/test_provider_support.py +++ b/tests/integration/megatron/model_support/test_provider_support.py @@ -147,6 +147,16 @@ def test_qwen38_27b_uses_qwen35_dense_support() -> None: assert handler.key == "qwen3_5_dense" +def test_qwen35_9b_uses_qwen35_dense_support() -> None: + spec = get_model_support_spec("Qwen/Qwen3.5-9B") + handler = get_model_support_handler("Qwen/Qwen3.5-9B") + + assert spec.key == "qwen3_5_dense" + assert spec.is_moe is False + assert spec.native_vllm_lora_status == "validated" + assert handler.key == "qwen3_5_dense" + + def test_glm53_bf16_uses_glm52_support() -> None: spec = get_model_support_spec("zai-org/GLM-5.3-BF16") handler = get_model_support_handler("zai-org/GLM-5.3-BF16")