[Model] Support DeepSeek-V4
This commit is contained in:
12
vllm_mlu/model_executor/models/__init__.py
Executable file
12
vllm_mlu/model_executor/models/__init__.py
Executable file
@@ -0,0 +1,12 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-MLU project
|
||||
|
||||
from vllm import ModelRegistry
|
||||
|
||||
|
||||
def register_model():
|
||||
from .deepseek_v4 import MLUDeepseekV4ForCausalLM # noqa: F401
|
||||
|
||||
ModelRegistry.register_model(
|
||||
"DeepseekV4ForCausalLM",
|
||||
"vllm_mlu.model_executor.models.deepseek_v4:MLUDeepseekV4ForCausalLM")
|
||||
Reference in New Issue
Block a user