From 0cd36081c50b4fefaa5f6a05f103cd6e5645a702 Mon Sep 17 00:00:00 2001 From: Tuukka Sarvi Date: Wed, 5 Aug 2026 08:32:29 +0000 Subject: [PATCH] [Bugfix][ROCm] Fix DeepSeek V4 DSpark probabilistic startup Signed-off-by: Tuukka Sarvi --- vllm/models/deepseek_v4/amd/dspark.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/vllm/models/deepseek_v4/amd/dspark.py b/vllm/models/deepseek_v4/amd/dspark.py index 4769483ff279..28056cb550db 100644 --- a/vllm/models/deepseek_v4/amd/dspark.py +++ b/vllm/models/deepseek_v4/amd/dspark.py @@ -290,6 +290,8 @@ class DSparkDeepseekV4ForCausalLM(nn.Module): # load_dspark_model always aliases the target's. has_own_embed_tokens = False has_own_lm_head = False + # Full-vocab draft: draft ids are target ids, no remapping needed. + draft_id_to_target_id = None def __init__(self, *, vllm_config: VllmConfig, prefix: str = "") -> None: super().__init__()