[Bugfix][ROCm] Fix DeepSeek V4 DSpark probabilistic startup (#51145)
Signed-off-by: Tuukka Sarvi <tuukka.sarvi@amd.com>
Tuukka Sarvi committed
Aug 11, 2026 at 17:15 UTC
12bea3eedcdf8e049f2478739b45ccb1456c03c0
1 file changed
+2
vllm/models/deepseek_v4/amd/dspark.py
+2
@@ -290,6 +290,8 @@ class DSparkDeepseekV4ForCausalLM(nn.Module):
290
# load_dspark_model always aliases the target's.
291
has_own_embed_tokens = False
292
has_own_lm_head = False
293
+ # Full-vocab draft: draft ids are target ids, no remapping needed.
294
+ draft_id_to_target_id = None
295
296
def __init__(self, *, vllm_config: VllmConfig, prefix: str = "") -> None:
297
super().__init__()