执行摘要

• 一句话：修复 LTX2 启用 cache-dit 时因 block 缺少 idx 属性导致的崩溃
• 推荐动作：本 PR 建议精读，它展示了一个典型的“包装对象丢失原始属性”问题的修复模式。使用 getattr 安全回退的方式简单有效，但需注意默认值的语义影响。对于依赖 idx 来精确控制 skip/perturbation 的用户可能需要额外配置。未来的改进可以考虑枚举索引以保持功能完整。

功能与动机

Issue #23193 报告 LTX2 启用 cache-dit 时抛出 AttributeError: 'CachedBlocks_Pattern_0_1_2' object has no attribute 'idx'。作者发现根因是 cache 机制包装了 blocks，导致它们不再拥有 idx 属性。PR 旨在恢复 LTX2 对 cache-dit 的支持。

实现拆解

1. 安全获取 block 索引：在 python/sglang/multimodal_gen/runtime/models/dits/ltx_2.py 的 forward 方法中，在 for 循环体开头新增 block_idx = getattr(block, 'idx', -1)，避免直接访问 block.idx 抛 AttributeError。
2. 替换所有 block.idx 引用：将循环内原本使用 block.idx 的 6 处（两处 skip 判断和四处 _ltx2_batched_perturbation_mask 调用）全部替换为 block_idx。
3. 保持兼容性：此改动不影响未启用 cache-dit 时的行为（此时 block 有 idx 属性），同时使得 cache-dit 启用时 block_idx 回退为 -1，不会误加入任何 skip 或 perturbation 集合，从而避免功能异常。
4. 配套测试：无新增测试，但作者提供了启用/禁用 cache-dit 时的可视化输出和端到端耗时对比（44.90s vs 22.44s），验证功能正常和性能提升。

关键文件：

• python/sglang/multimodal_gen/runtime/models/dits/ltx_2.py（模块 扩散模型；类别 source；类型 data-contract；符号 forward）: 包含所有变更：修复 block.idx 访问问题，是 PR 的唯一修改文件。

关键符号：forward

关键源码片段

python/sglang/multimodal_gen/runtime/models/dits/ltx_2.py

包含所有变更：修复 block.idx 访问问题，是 PR 的唯一修改文件。

# ltx_2.py forward 方法关键片段
# 安全获取 block 索引，避免 cache 包装对象缺少 idx 属性
for block in self.transformer_blocks:
    # 使用 getattr 安全获取 idx，当 cache-dit 启用时返回 -1
    block_idx = getattr(block, "idx", -1)
    # 后续所有 block.idx 引用替换为 block_idx
    skip_video_self_attn = block_idx in skip_video_self_attn_blocks
    skip_audio_self_attn = block_idx in skip_audio_self_attn_blocks
    # ... 扰动掩码调用同样使用 block_idx

评论区精华

• gemini-code-assist[bot] 指出初始实现中 getattr(block, 'idx', -1) 将 block.idx 的赋值放在了 for 循环之前，这会导致 NameError（block 未定义）。建议改用 enumerate 来自动提供索引，同时指出默认值 -1 会禁用 skip/perturbation 特性。
• ping1jing2 提醒作者注意 gemini 的评论。

• gjsheu（PR 作者）回复“done”，表示已修复。然而最终合并的代码仍然是循环内 getattr 的写法，并未采用 enumerate 方案。review 人员 mickqian 最终批准了合并。

• block 变量作用域与 getattr 放置位置 (correctness): 作者接受建议，将获取移至循环体内，但最终未采用 enumerate 方案，仍保留 getattr 在循环内。

• 默认值 -1 对 skip/perturbation 功能的影响 (design): 作者未采用 enumerate，但合并时被批准。本质上这是一个设计权衡：接受 cache 启用时 skip/perturbation 失效，换取最小化改动。

风险与影响

• 风险：

  1. 核心路径变更：改动位于 diffusion 模型核心 forward 函数，影响所有对 transformer_blocks 的迭代，但改动极小，回退方案安全。
  2. 缺少测试覆盖：没有对应的单元测试验证 cache-dit 启用/禁用场景。
  3. 回退语义：使用 -1 作为默认索引意味着当 cache-dit 启用时，所有 skip/perturbation 功能都会失效（不会被跳过也不会被扰动），而不仅仅是无法读取原始 idx。这可能与用户期望不符，但作者提供了可视化结果证明原始输出质量不受影响。
     - 影响：用户：LTX2 用户现可启用 cache-dit 获得约 2 倍端到端加速（44.90s → 22.44s），并且输出质量与禁用 cache 时视觉上接近。
     系统：仅影响 LTX2 模型，不涉及其他 diffusion 模型或通用推理路径。
     团队：无运维负担，改动侵入性极小。
     影响程度：低风险、高收益的针对性 bugfix。

• 风险标记：核心路径变更, 缺少测试覆盖, 回退语义可能影响功能完整性

关联脉络

• PR #2070 相关 PR（issue 中提及导致问题的原始 PR）: Issue #23193 指出该 bug 可能与 PR #2070 的合并有关，该 PR 引入了 cache-dit 支持，改变了 block 对象结构。