21 lines
625 B
Python
21 lines
625 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""DeepSeek V4.1 hardware-isolated model entry point."""
|
|
|
|
from vllm.platforms import current_platform
|
|
|
|
from .quant_config import DeepseekV4FP8Config
|
|
|
|
if current_platform.is_rocm():
|
|
from .amd.dspark import DSparkDeepseekV4ForCausalLM
|
|
from .amd.vl_model import DeepseekV41ForCausalLM
|
|
else:
|
|
from .nvidia.dspark import DSparkDeepseekV4ForCausalLM
|
|
from .nvidia.vl_model import DeepseekV41ForCausalLM
|
|
|
|
__all__ = [
|
|
"DSparkDeepseekV4ForCausalLM",
|
|
"DeepseekV4FP8Config",
|
|
"DeepseekV41ForCausalLM",
|
|
]
|