[Bugfix][Spec Decode] Add missing draft_id_to_target_id to DSparkDeepseekV4ForCausalLM (#47429)
Signed-off-by: Laurent-Zhang <zhangdongsheng80@gmail.com> Co-authored-by: Claude <noreply@anthropic.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
Laurent-Zhang committed
Jul 6, 2026 at 22:55 UTC
8d8ec383619d956b25611a3312a431c9a6ca1c04
1 file changed
+2
vllm/models/deepseek_v4/nvidia/dspark.py
+2
@@ -269,6 +269,8 @@ class DSparkDeepseekV4ForCausalLM(nn.Module):
269
# load_dspark_model always aliases the target's.
270
has_own_embed_tokens = False
271
has_own_lm_head = False
272
+ # Full-vocab draft: draft ids are target ids, no remapping needed.
273
+ draft_id_to_target_id = None
274
275
def __init__(self, *, vllm_config: VllmConfig, prefix: str = "") -> None:
276
super().__init__()