#20706 [diffusion] Unify `TeaCacheParams` and `WanTeaCacheParams`
作者 eitanturok · 合并时间 2026-03-28 09:51
统一TeaCache参数类,移除Wan特定实现,支持动态系数回调。
建议工程师精读此PR,重点关注: - TeaCacheParams类的设计变化,特别是get_coefficients方法和coefficients_callback机制,展示了如何通过回调支持动态系数选择。 - CFG逻辑的简化处理(get_skip_boundaries方法),这对理解扩散模型中的缓存优化和步骤管理有参考价值。 - 单元测试的设计,验证兼容性和回调优先级,可作为类似重构的测试范例。
参与讨论