EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

clean up model module (#4611)

Browse Source 
Model module is useless now. Let't remove it totally.

- vLLM version: v0.11.2

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
wangxiyuan
2025-12-02 17:35:47 +08:00
committed by GitHub
parent 96b2cdf6d8
commit 874097a1de
3 changed files with 0 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@@ -534,7 +534,6 @@ setup(
entry_points={ entry_points={
"vllm.platform_plugins": ["ascend = vllm_ascend:register"], "vllm.platform_plugins": ["ascend = vllm_ascend:register"],
"vllm.general_plugins": [ "vllm.general_plugins": [
"ascend_enhanced_model = vllm_ascend:register_model",
"ascend_kv_connector = vllm_ascend:register_connector", "ascend_kv_connector = vllm_ascend:register_connector",
"ascend_model_loader = vllm_ascend:register_model_loader", "ascend_model_loader = vllm_ascend:register_model_loader",
"ascend_service_profiling = vllm_ascend:register_service_profiling" "ascend_service_profiling = vllm_ascend:register_service_profiling"

5
vllm_ascend/__init__.py
View File

@@ -22,11 +22,6 @@ def register():
return "vllm_ascend.platform.NPUPlatform" return "vllm_ascend.platform.NPUPlatform"
def register_model():
from .models import register_model
register_model()
def register_connector(): def register_connector():
from vllm_ascend.distributed import register_connector from vllm_ascend.distributed import register_connector
register_connector() register_connector()

10
vllm_ascend/models/__init__.py
View File

@@ -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"
)