ComfyUI v0.45.0 将 Qwen3.x 文本生成速度提升一倍

ComfyUI Wikinews

ComfyUI v0.45.0 重写了 Qwen3.x 文本生成:Qwen3.5 9B 在 RTX 5090 上从每秒 106 个 token 提升到 219 个,并引入自适应树形 MTP 与 INT8 注意力。

Comfy Org 于 11 月 11 日发布了一篇 ComfyUI 工程博客。Qwen3.x 的文本生成路径经过重写,使得相同的模型在同一块 GPU 上运行速度最高提升 2 倍,而模型本身及其生成的文本都没有任何改变。文中介绍的全部内容都随 ComfyUI v0.45.0 发布。

ComfyUI 中的提示词增强、图像描述和故事写作都属于文本生成,因此都会获得这次提速。唯一需要设置的是 Generate Text 节点的 mtp 选项,它默认保持为 auto。

ComfyUI v0.45.0 文本生成重写后的 token 速率

摘自 Comfy-Org 宣布此次重写的工程博客。

实测 token 速率

模型GPU之前v0.45.0
Qwen3.5 9BRTX 5090106 tok/s219 tok/s
Qwen3.5 9BRTX 508091 tok/s129 tok/s
Qwen3.8 27B,描述一张 1536×1536 的图像RTX 509055 tok/s122 tok/s

在 RTX 5090 上,Qwen3.8 27B W4A8 在描述一张 1500×1500 的图像并撰写一个故事时,达到了每秒 137 个 token。

有哪些变化

  • 矩阵-向量内核:W4A8 和 W6A8 权重现在按照内核读取的顺序与各自的按组缩放系数交错排列,因此每个线程块流式读取一段连续内存,而不必在两个 tensor 之间来回跳转。重新布局在每次模型加载时于显存中执行一次,不额外占用内存或显存。
  • 自适应树形 MTP:多 token 预测头不再只起草单条猜测链。它把次优候选 token 保留为同级分支,在一次遍历中最多验证八行,并每 32 步重新规划树的形状。代码、列表以及输出中复用的提示词更适合深度;创意写作更适合广度。
  • Comfy-Kitchen 注意力:为 LLM 注意力路径加入 INT8 注意力并支持 KV 缓存,从而减少 KV 缓存占用的显存,并加快长提示词和系统提示词上的预填充速度。
ComfyUI v0.45.0 中的链式 MTP 与树形 MTP 对比

链式 MTP 只起草一条路径;树形 MTP 把次优候选 token 保留为同级分支,并将它们一起验证。

如何启用

  1. 更新到 ComfyUI v0.45.0 或更新版本。
  2. 加载一个 Qwen3.x 模型,并通过 Generate Text 节点运行。
  3. 让 mtp 保持为 auto。固定值 2 到 5 仍然会固定起草深度,而 tree* 模式会同时固定广度和深度。
  4. 可选:以 --use-ck-attention 启动 ComfyUI 以启用 Comfy-Kitchen 注意力。这需要一个包含注意力支持的 Comfy Kitchen 构建版本。

以上所有功能在 Windows 和 Linux 上的 NVIDIA 与 AMD GPU 均可使用,过去会因显存不足而失败的长上下文现在也能正常运行。节点提示框中还有一个细节:对于相同的种子,MTP 输出的分布仍然正确,但与不使用 MTP 的输出并不相同。

评论

使用 GitHub 登录后即可参与讨论。

评论加载中…
ComfyUI v0.45.0 将 Qwen3.x 文本生成速度提升一倍 | ComfyUI Wiki