From 874097a1de137cbed712bf362c85200a9d2a19b7 Mon Sep 17 00:00:00 2001 From: wangxiyuan Date: Tue, 2 Dec 2025 17:35:47 +0800 Subject: [PATCH] clean up model module (#4611) Model module is useless now. Let't remove it totally. - vLLM version: v0.11.2 Signed-off-by: wangxiyuan --- setup.py | 1 - vllm_ascend/__init__.py | 5 ----- vllm_ascend/models/__init__.py | 10 ---------- 3 files changed, 16 deletions(-) delete mode 100644 vllm_ascend/models/__init__.py diff --git a/setup.py b/setup.py index 3e88affa..890b5228 100644 --- a/setup.py +++ b/setup.py @@ -534,7 +534,6 @@ setup( entry_points={ "vllm.platform_plugins": ["ascend = vllm_ascend:register"], "vllm.general_plugins": [ - "ascend_enhanced_model = vllm_ascend:register_model", "ascend_kv_connector = vllm_ascend:register_connector", "ascend_model_loader = vllm_ascend:register_model_loader", "ascend_service_profiling = vllm_ascend:register_service_profiling" diff --git a/vllm_ascend/__init__.py b/vllm_ascend/__init__.py index 8c8bd514..117859df 100644 --- a/vllm_ascend/__init__.py +++ b/vllm_ascend/__init__.py @@ -22,11 +22,6 @@ def register(): return "vllm_ascend.platform.NPUPlatform" -def register_model(): - from .models import register_model - register_model() - - def register_connector(): from vllm_ascend.distributed import register_connector register_connector() diff --git a/vllm_ascend/models/__init__.py b/vllm_ascend/models/__init__.py deleted file mode 100644 index b1957fe8..00000000 --- a/vllm_ascend/models/__init__.py +++ /dev/null @@ -1,10 +0,0 @@ -from vllm import ModelRegistry - - -def register_model(): - # 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. - ModelRegistry.register_model( - "PanguProMoEForCausalLM", - "vllm_ascend.torchair.models.torchair_pangu_moe:PanguProMoEForCausalLM" - )