Prhub

#32639 docs: clarify diffusion stage reuse guidance

原始 PR 作者 mickqian 合并时间 2026-07-28 19:52 文件变更 1 提交数 2 评论 2 代码增减 +111 / -18

执行摘要

阐明扩散模型 stage 复用推荐顺序

PR body 指出目标是"keep new model ports smaller, faster to review, and closer to shared optimized runtime paths",避免将 aggregated custom stage 作为默认选择,从而更好地利用 native stage 带来的 offload、组件就绪、profile、disaggregation、batching 等优化。

建议在 diffusion 新模型开发前精读此文档,尤其是 stage 形状选择阶梯;评审者可快速确认贡献者是否遵循了推荐顺序。

实现拆解

  1. 在表格中重新排列 stage 形状的优先级顺序,将 Native stages 和 Native-stage subclass 提前,Aggregated custom stage 降为最后选择。
  2. 增加更细致的阶梯分类(native-stage subclass、custom single-purpose stage),并补充每种形状的适用条件和布局示例。
  3. 更新 pipeline 接线示例:将自定义 prompt 处理改为继承 TextEncodingStage,并在周边保留 native stage;把 aggregated BeforeDenoisingStage 标记为最后手段。
  4. 调整入门文档中的行为说明,使 stage 选择与架构目标更一致。
文件 模块 状态 重要度
docs_new/docs/sglang-diffusion/support_new_models.mdx 扩散指南 modified 6.22

关键符号

MyModelPipeline create_pipeline_stages

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

文档变更本身无直接运行风险,但若指导模糊或错误,可能误导新模型贡献者选择非最优 stage 形状,影响后期维护性和性能。本次 PR 提供了明确分级,风险已降低。

影响范围主要为后续添加 diffusion 模型时参考此文档的开发者,引导其采用更合理 stage 复用策略,减少不必要的自定义 stage,从而提升代码共享度和审查效率。无用户可见影响。

指导性变更需确保准确

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论