[Misc] Remove pangu model file (#2798)
vllm-ascend won't contain model file anymore. Now pangu model file has
been moved to torchair module. The origin one can be removed.
Note: After this PR, pangu only works with torchair mode then.
- vLLM version: v0.10.1.1
- vLLM main:
8c892b1831
Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
@@ -4,15 +4,6 @@ import vllm_ascend.envs as envs_ascend
|
|||||||
|
|
||||||
|
|
||||||
def register_model():
|
def register_model():
|
||||||
from .deepseek_dbo import CustomDeepseekDBOForCausalLM # noqa: F401
|
|
||||||
from .deepseek_mtp import CustomDeepSeekMTP # noqa: F401
|
|
||||||
from .deepseek_v2 import CustomDeepseekV2ForCausalLM # noqa: F401
|
|
||||||
from .deepseek_v3 import CustomDeepseekV3ForCausalLM # noqa: F401
|
|
||||||
from .qwen2_5_vl import \
|
|
||||||
AscendQwen2_5_VLForConditionalGeneration # noqa: F401
|
|
||||||
from .qwen2_vl import AscendQwen2VLForConditionalGeneration # noqa: F401
|
|
||||||
from .qwen3 import CustomQwen3ForCausalLM # noqa: F401
|
|
||||||
|
|
||||||
ModelRegistry.register_model(
|
ModelRegistry.register_model(
|
||||||
"DeepSeekMTPModel",
|
"DeepSeekMTPModel",
|
||||||
"vllm_ascend.models.deepseek_mtp:CustomDeepSeekMTP")
|
"vllm_ascend.models.deepseek_mtp:CustomDeepSeekMTP")
|
||||||
@@ -56,6 +47,9 @@ def register_model():
|
|||||||
ModelRegistry.register_model(
|
ModelRegistry.register_model(
|
||||||
"Qwen3ForCausalLM", "vllm_ascend.models.qwen3:CustomQwen3ForCausalLM")
|
"Qwen3ForCausalLM", "vllm_ascend.models.qwen3:CustomQwen3ForCausalLM")
|
||||||
|
|
||||||
|
# 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(
|
ModelRegistry.register_model(
|
||||||
"PanguProMoEForCausalLM",
|
"PanguProMoEForCausalLM",
|
||||||
"vllm_ascend.models.pangu_moe:PanguProMoEForCausalLM")
|
"vllm_ascend.torchair.models.torchair_pangu_moe:PanguProMoEForCausalLM"
|
||||||
|
)
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user