diff --git a/tests/ut/torchair/test_utils.py b/tests/ut/torchair/test_utils.py index ab85d28..8aebb9d 100644 --- a/tests/ut/torchair/test_utils.py +++ b/tests/ut/torchair/test_utils.py @@ -70,7 +70,7 @@ class TestTorchairUtils(TestBase): ), ("Qwen2ForCausalLM", "vllm_ascend.torchair.models.qwen2:CustomQwen2ForCausalLM"), - ("Qwen3ForCausalLM", + ("Qwen3MoeForCausalLM", "vllm_ascend.torchair.models.qwen3_moe:CustomQwen3MoeForCausalLM") ] diff --git a/vllm_ascend/torchair/utils.py b/vllm_ascend/torchair/utils.py index 8448ddc..2f3a28e 100644 --- a/vllm_ascend/torchair/utils.py +++ b/vllm_ascend/torchair/utils.py @@ -168,7 +168,7 @@ def register_torchair_model(): "vllm_ascend.torchair.models.qwen2:CustomQwen2ForCausalLM") ModelRegistry.register_model( - "Qwen3ForCausalLM", + "Qwen3MoeForCausalLM", "vllm_ascend.torchair.models.qwen3_moe:CustomQwen3MoeForCausalLM")