11 lines
424 B
Python
11 lines
424 B
Python
from vllm import ModelRegistry
|
|
|
|
|
|
def register_model():
|
|
ModelRegistry.register_model("DeepseekV4ForCausalLM", "vllm_ascend.models.deepseek_v4:AscendDeepseekV4ForCausalLM")
|
|
|
|
ModelRegistry.register_model("DeepSeekV4MTPModel", "vllm_ascend.models.deepseek_v4_mtp:DeepSeekV4MTP")
|
|
ModelRegistry.register_model(
|
|
"LlamaForCausalLMVwnEagle3", "vllm_ascend.models.llama_eagle3_vwn:Eagle3VwnLlamaForCausalLM"
|
|
)
|