Add diffusion backend & Wan training support#779

Merged

wenxie-amd merged 20 commits into

mainfrom

Jul 8, 2026

Commits on Jun 17, 2026

Commits on Jun 18, 2026

fix(diffusion): broadcast Wan timesteps per sample

zirui
and
cursoragent
committed

Commits on Jun 30, 2026

Remove unused pip packages and clean up README.md
zirui
committed

Commits on Jul 1, 2026

Commits on Jul 6, 2026

fix: add missing newline at end of examples/diffusion/README.md
Copilot
authored

Commits on Jul 7, 2026

set aiter as default attention backend & add set seperated_timestep=True for wan2.2-5B
zirui
committed
fix(diffusion): address Wan code quality findings

zirui
and
cursoragent
committed
fix(diffusion): resolve remaining code quality findings

zirui
and
cursoragent
committed
fix(diffusion): address Copilot PR review findings

zirui
and
cursoragent
committed
style(diffusion): apply pre-commit formatting

zirui
and
cursoragent
committed
fix(diffusion): address remaining PR review comments

zirui
and
cursoragent
committed
fix(diffusion): align Wan timestep config naming

zirui
and
cursoragent
committed
fix(diffusion): resolve Wan training review issues

zirui
and
cursoragent
committed
Fix black formatting in train_pipeline.py to pass code-lint CI
Copilot
authored
fix(diffusion): remove Wan attention debug leftovers

zirui
and
cursoragent
committed
fix(diffusion): resolve Wan review follow-ups

zirui
and
cursoragent
committed
fix(diffusion): address latest Wan review comments

zirui
and
cursoragent
committed