全词表 EAGLE3 草稿跳过冗余重映射,对齐 dspark 契约
值得精读,但只需 10 分钟:改动只有 24 行且集中在三处结构相同的构造逻辑。建议重点关注三个设计点——(1) 用 `None` 作为身份映射的 sentinel,把判断从热路径挪到构造期,是这类「参数化语义」的经典手法;(2) 比较对象选择 `vllm_config.model_config.get_vocab_size()` 而非草稿配置自身,这个选择决定了哪些真实模型能命中快速路径;(3) 维护者删除测试的权衡是否可接受。若后续要控制维护成本,更经济的做法不是放弃守卫,而是把这段条件逻辑抽成一个共享 helper(例如 `maybe_create_draft_id_to_target_id()`),在 eagle3 / dspark / dflash 之间复用,既消除四处重复代码,也能用单测覆盖一个函数而不是四个模型。阅读时另需注意本 PR 由 AI 辅助(Claude Code / GLM-5.3)编写、作者自审,且关键契约确认(@benchislett)未获回复。
参与讨论