diffsynth-pipeline-training
Installation
SKILL.md
DiffSynth-Studio: Pipeline 训练接入
为已接入推理 Pipeline 的模型系列创建训练模块并完成训练验证。整个系列共用一个训练脚本,一次调用完成全部训练模块的创建和验证。
⚠️ 核心约束:只改训练相关 Unit
训练接入时,只能修改与训练直接相关的 Unit(如 InputXXXEmbedder 的 pipe.scheduler.training 分支、新增训练专用 Unit),绝对禁止修改 Pipeline 的其他任何内容。
推理 Pipeline 已经过 E2E 测试验证,任何对非训练相关代码的改动都可能破坏已做好的推理流程。
| 允许修改 | 禁止修改 |
|---|---|
InputXXXEmbedder 的 pipe.scheduler.training 分支逻辑 |
DiT/VAE/TextEncoder 等模型组件 |
新增 InputXXXEmbedder Unit(如缺失) |
推理用 Unit 的 pipe.scheduler.training=False 分支 |
训练脚本(train.py、.sh 配置、验证脚本) |
Pipeline 的 Unit 链顺序、__call__ 逻辑 |
get_pipeline_inputs 映射逻辑 |
推理示例脚本 |
判断标准:如果改动的代码只在 pipe.scheduler.training=True 时执行,可以改;如果改动会影响 pipe.scheduler.training=False 的推理路径,不要改。