Skip to content
Merged
488 changes: 0 additions & 488 deletions tests/ut/models/test_qwen2_5_vl.py

This file was deleted.

422 changes: 0 additions & 422 deletions tests/ut/models/test_qwen2_5_vl_without_padding.py

This file was deleted.

19 changes: 2 additions & 17 deletions vllm_ascend/models/__init__.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
from vllm import ModelRegistry

import vllm_ascend.envs as envs_ascend


def register_model():
ModelRegistry.register_model(
Expand All @@ -10,24 +8,11 @@ def register_model():

ModelRegistry.register_model(
"Qwen3VLMoeForConditionalGeneration",
"vllm_ascend.models.qwen2_5_vl_without_padding:AscendQwen3VLMoeForConditionalGeneration"
)
"vllm_ascend.models.qwen3_vl:AscendQwen3VLMoeForConditionalGeneration")

ModelRegistry.register_model(
"Qwen3VLForConditionalGeneration",
"vllm_ascend.models.qwen2_5_vl_without_padding:AscendQwen3VLForConditionalGeneration"
)

if envs_ascend.USE_OPTIMIZED_MODEL:
ModelRegistry.register_model(
"Qwen2_5_VLForConditionalGeneration",
"vllm_ascend.models.qwen2_5_vl:AscendQwen2_5_VLForConditionalGeneration"
)
else:
ModelRegistry.register_model(
"Qwen2_5_VLForConditionalGeneration",
"vllm_ascend.models.qwen2_5_vl_without_padding:AscendQwen2_5_VLForConditionalGeneration_Without_Padding"
)
"vllm_ascend.models.qwen3_vl:AscendQwen3VLForConditionalGeneration")

# There is no PanguProMoEForCausalLM in vLLM, so we should register it before vLLM config initialization
# to make sure the model can be loaded correctly. This register step can be removed once vLLM support PanguProMoEForCausalLM.
Expand Down
Loading
Loading