新 ComfyUI 节点砍 Qwen3 编码器显存 5.5→3.2GB,输出逐位一致

Occsan · reddit · 2026-10-28

Flux2 Klein 实际只读取其 Qwen3 文本编码器的 3 个隐藏层(第 9、18、27 层),但标准加载器会构建并运行完整的 36 层模型加上最终 norm 和 LM head,且 ComfyUI-GGUF 加载时会把整个 token embedding 表反量化为 fp16。作者写了「Klein CLIP Loader (Pruned)」节点,只构建实际需要的部分:

实测(Qwen3-8B Q4KM,Klein 9B,RTX 4090,512-token 提示词):TE 权重占显存 5.51→3.20 GB,编码峰值 5.97→3.62 GB,为 DiT/LoRA/旁挂 LLM 腾出约 2.3 GB——对 8GB 显卡可能决定是否需要 offload。安装:clone 仓库到 customnodes 后换用该加载器即可,仅需 ComfyUI-GGUF。

另外作者 profiling 时发现 ComfyUI 按引用数和大小而非重载耗时选择卸载模型,已提 PR 改为按实测重载时间排序,Klein+本地 LLM 工作流每轮 13.6s→6.7s(未合并)。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →