23 lines
809 B
Python
23 lines
809 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
"""DeepSeek V3.2 (``deepseek_v32``) model — hardware-isolated entry point.
|
|
|
|
DeepSeek V3.2 introduced the DeepSeek Sparse Attention (DSA) architecture:
|
|
MLA + a "lightning indexer" that selects the top-k tokens for a sparse MLA
|
|
attend. The same model code serves any DSA checkpoint, including GLM-5.2
|
|
(``glm_moe_dsa``), which reuses this architecture.
|
|
"""
|
|
|
|
from vllm.platforms import current_platform
|
|
|
|
if current_platform.is_rocm() or current_platform.is_xpu():
|
|
raise NotImplementedError("deepseek_v32 currently supports NVIDIA SM100 only.")
|
|
|
|
from .nvidia.model import DeepseekV32ForCausalLM
|
|
from .nvidia.mtp import DeepseekV32MTP
|
|
|
|
__all__ = [
|
|
"DeepseekV32ForCausalLM",
|
|
"DeepseekV32MTP",
|
|
]
|