main
py 14 lines 537 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3 """Stub for platforms where the DeepSeek-V4 vision variant is unsupported."""
4
5 from torch import nn
6
7
8 class DeepseekV4ForConditionalGeneration(nn.Module):
9 def __init__(self, *, vllm_config, prefix: str = ""):
10 super().__init__()
11 raise NotImplementedError(
12 "DeepSeek-V4 vision (DeepseekV4ForConditionalGeneration) is not "
13 "supported on the current platform; use CUDA or ROCm."
14 )