main
py 21 lines 753 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3 """Compatibility imports for the platform-neutral DeepSeek-V4 vision model."""
4
5 from ..common.vl_model import (
6 DeepseekV4ForConditionalGeneration,
7 )
8 from ..common.vl_model import (
9 _make_deepseek_v4_vl_weights_mapper as _make_common_vl_weights_mapper,
10 )
11 from .model import _make_deepseek_v4_weights_mapper
12
13
14 def _make_deepseek_v4_vl_weights_mapper(expert_dtype: str, image_enabled: bool):
15 """Retain the original NVIDIA helper signature for downstream imports."""
16 return _make_common_vl_weights_mapper(
17 _make_deepseek_v4_weights_mapper(expert_dtype), image_enabled
18 )
19
20
21 __all__ = ["DeepseekV4ForConditionalGeneration"]