Prhub

#24682 [diffusion] doc: update ltx2 multi-gpu deployment guide

原始 PR 作者 mickqian 合并时间 2026-05-08 18:38 文件变更 5 提交数 7 评论 3 代码增减 +76 / -18

执行摘要

更新 LTX 多 GPU 部署文档

原 LTX 文档仅提供单 GPU 部署示例,缺少多 GPU 并行部署的指导。用户需要了解如何使用 2 或 4 GPU 运行 LTX,以及如何正确配置 --num-gpus、--enable-cfg-parallel 和 --tp-size 参数。

对于使用 LTX 模型的用户,建议仔细阅读新增的多 GPU 预设部分。对于前端组件维护者,可参考 getParallelFlags 的对象查找模式。

讨论亮点

在 review 中,gemini-code-assist[bot] 提出两点优化建议:将硬件判断改为 startsWith('h200') 以提高可扩展性;将条件分支重构为对象查找 parallelFlagsMap 以提高可维护性。开发者采纳了这些建议,最终代码按此方式实现。

实现拆解

  1. 扩展硬件选项: 在 docs_new/src/snippets/diffusion/ltx-deployment.jsx 中新增 h200-2gpuh200-4gpu 硬件选项,并实现 getParallelFlags() 函数,根据硬件选择生成对应的命令行参数(--num-gpus 2 --enable-cfg-parallel--num-gpus 4 --tp-size 2 --enable-cfg-parallel)。

  2. 重命名 Cookbook 页面: 将 LTX.mdx 重命名为 LTX2 & LTX2.3.mdx,更新页面标题和内容,新增“Fast multi-GPU presets”表格,推荐不同目标下的启动参数。

  3. 更新导航配置: 在 docs_new/docs.json 中将 LTX 分组页面路径指向新的文件名。

  4. 添加徽标资源: 新增 ltx.svg 文件,在介绍页面使用该图标替换占位卡片图。

  5. 更新入口页面: 在 intro.mdx 中更新 LTX 卡片链接和图片为新路径和新徽标。

文件 模块 状态 重要度
docs_new/src/snippets/diffusion/ltx-deployment.jsx 交互组件 modified 6.77
docs_new/cookbook/diffusion/LTX/LTX2 & LTX2.3.mdx 用户指南 renamed 5.15
docs_new/docs.json 导航配置 modified 2.5
docs_new/cookbook/diffusion/intro.mdx 概览页 modified 2.49
docs_new/cards/logos/ltx.svg 资源文件 added 2.76

关键符号

getParallelFlags

关键源码片段

docs_new/src/snippets/diffusion/ltx-deployment.jsx core-logic

核心交互组件,新增多 GPU 硬件选项和参数生成逻辑

const getParallelFlags = () => {
  // 使用对象查找维护并行配置,方便后续新增硬件选项
  const parallelFlagsMap = {
    'h200-2gpu': ` --num-gpus 2 --enable-cfg-parallel`,
    'h200-4gpu': ` --num-gpus 4 --tp-size 2 --enable-cfg-parallel`,
  };
  return parallelFlagsMap[values.hardware] || '';
};const getDeviceMode = () => {
  // 使用 startsWith 匹配所有 h200-* 前缀,提升可扩展性
  if (values.hardware.startsWith('h200')) {
    return 'resident';
  }
  // ...
};

评论区精华

简化硬件判断为 startsWith 设计

gemini-code-assist[bot] 建议用 startsWith('h200') 替代多个 '===' 判断。

结论:已采纳,代码改为 startsWith · 已解决

重构为对象查找 map 设计

gemini-code-assist[bot] 建议将多个 if 改为对象查找 parallelFlagsMap

结论:已采纳,最终代码使用对象查找 · 已解决

风险与影响

文档变更主要风险在于命令行参数可能错误,但经过审查已降低。不涉及代码逻辑变更,无回归风险。

对用户:提供清晰的多 GPU 部署指引,降低配置错误概率。对团队:简化后续新增硬件配置的维护。

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论