| 1 | # SPDX-License-Identifier: Apache-2.0 |
| 2 | # SPDX-FileCopyrightText: Copyright contributors to the vLLM project |
| 3 | """DeepSeek V4.1 hardware-isolated model entry point.""" |
| 4 | |
| 5 | from vllm.platforms import current_platform |
| 6 | |
| 7 | from .quant_config import DeepseekV4FP8Config |
| 8 | |
| 9 | if current_platform.is_rocm(): |
| 10 | from .amd.dspark import DSparkDeepseekV4ForCausalLM |
| 11 | from .amd.vl_model import DeepseekV41ForCausalLM |
| 12 | else: |
| 13 | from .nvidia.dspark import DSparkDeepseekV4ForCausalLM |
| 14 | from .nvidia.vl_model import DeepseekV41ForCausalLM |
| 15 | |
| 16 | __all__ = [ |
| 17 | "DSparkDeepseekV4ForCausalLM", |
| 18 | "DeepseekV4FP8Config", |
| 19 | "DeepseekV41ForCausalLM", |
| 20 | ] |