执行摘要

• 一句话：修复 DSpark 并行草稿 token ID 初始化缺少回退字段
• 推荐动作：值得合入。改动小且无争议，但建议在合并前或随后补一个单测，确认 mask_token_id 优先级不会影响其他并行草稿模型；若需深挖，可以对照 MRV2 的 get_parallel_drafting_token_id 实现，评估是否适合抽取公共解析函数。

功能与动机

PR body 明确指出："Fix dspark parallel_drafting_token_id init bug in MRV1. The bug has been fixed in MRV2 already in get_parallel_drafting_token_id function." 也就是说，MRV2 已在 get_parallel_drafting_token_id 中修复了相同问题，本次是回迁修复到 MRV1，使 DSpark 草稿模型在 MRV1 下也能正确初始化。

实现拆解

1. 修改 vllm/v1/spec_decode/llm_base_proposer.py 的 _init_parallel_drafting_params：在原 dflash_config.mask_token_id 分支后优先识别顶层 mask_token_id（非 None）与 dspark_noise_token_id，最后才回落到 pard_token / ptd_token_id，使 DSpark 模型可正常初始化。
2. 同步更新 ValueError 文案，列出全部五个可接受字段，便于后续配置排查。
3. 本 PR 未附带测试文件；建议在 tests/v1/spec_decode/ 补充针对 mask_token_id 与 dspark_noise_token_id 的初始化单测，防止回归。

关键文件：

• vllm/v1/spec_decode/llm_base_proposer.py（模块 投机解码；类别 source；类型 core-logic；符号 _init_parallel_drafting_params）: 修复 MRV1 并行草稿初始化遗漏 DSpark 配置字段的问题，与 MRV2 的 get_parallel_drafting_token_id 行为对齐。

关键符号：_init_parallel_drafting_params

关键源码片段

vllm/v1/spec_decode/llm_base_proposer.py

修复 MRV1 并行草稿初始化遗漏 DSpark 配置字段的问题，与 MRV2 的 get_parallel_drafting_token_id 行为对齐。

# vllm/v1/spec_decode/llm_base_proposer.py
def _init_parallel_drafting_params(self):
    # 并行草稿下，被 mask 的 slot 需要使用特定的 token ID。
    # EAGLE + 并行草稿时，还需要为这些 slot 准备 hidden state。
    model_hf_config = self.draft_model_config.hf_config
​
    # DFlash 模型把 mask token 放在 dflash_config 里，优先级最高。
    dflash_config = getattr(model_hf_config, "dflash_config", None)
    if dflash_config and "mask_token_id" in dflash_config:
        self.parallel_drafting_token_id = dflash_config["mask_token_id"]
    # DSpark 等模型直接在顶层配置 mask_token_id / dspark_noise_token_id，
    # MRV1 之前漏掉了这两个字段，导致初始化直接抛 ValueError。
    elif getattr(model_hf_config, "mask_token_id", None) is not None:
        self.parallel_drafting_token_id = model_hf_config.mask_token_id
    elif hasattr(model_hf_config, "dspark_noise_token_id"):
        self.parallel_drafting_token_id = model_hf_config.dspark_noise_token_id
    # 其余已支持厂商：pard_token / ptd_token_id。
    elif hasattr(model_hf_config, "pard_token"):
        self.parallel_drafting_token_id = model_hf_config.pard_token
    elif hasattr(model_hf_config, "ptd_token_id"):
        self.parallel_drafting_token_id = model_hf_config.ptd_token_id
    else:
        raise ValueError(
            "For parallel drafting, the draft model config must have "
            "`dflash_config.mask_token_id`, `mask_token_id`, "
            "`dspark_noise_token_id`, `pard_token`, or `ptd_token_id` "
            "specified in its config.json."
        )
​
    # EAGLE 并行草稿还需要额外准备 hidden state tensor。
    if self.pass_hidden_states_to_model:
        self.parallel_drafting_hidden_state_tensor = torch.empty(
            self.hidden_size, dtype=self.dtype, device=self.device
        )

评论区精华

本 PR 没有真正的 review 评论。claude[bot] 因这是 fork PR 自动跳过了深入审查；维护者 benchislett 和 mgoin 均直接 APPROVED，没有提出设计或实现层面的疑问。

• 暂无高价值评论线程

风险与影响

• 风险：
  • 优先级微调：mask_token_id 现在比 pard_token / ptd_token_id 更早被读取。若某个模型同时配置了多个字段，最终使用的 token 可能与改动前不同；不过并行草稿模型通常只写一个字段，影响有限。
  • 缺少测试：改动直击初始化路径，但没有单测覆盖，后续重构容易回归。建议至少补一个针对 mask_token_id 和 dspark_noise_token_id 的单元测试。
  • 影响范围：仅影响 MRV1 的并行草稿初始化，不影响 MRV2 或普通（非并行草稿）解码；DSpark 用户在 MRV1 下的启动错误会被修复。
  • 影响：影响范围集中在 MRV1（V1 executor）的并行草稿初始化路径，受益对象是使用 DSpark 草稿模型的用户。对非并行草稿推理和其他模型无影响，对团队的意义在于消除 MRV1 与 MRV2 之间已知的配置解析差异，降低后续维护成本。
  • 风险标记：缺少测试覆盖, 配置优先级变化, 仅 V1 路径

关联脉络

• PR #50734 [Bugfix][Model] Fix Qwen3.5 MTP for text-only checkpoints: 同为 MRV1 speculative-decoding 配置兼容性修复，可对照该功能线的演进。