Add flux.1 to diffusion backend#832

Merged

wenxie-amd merged 37 commits into

mainfrom

Jul 16, 2026

Commits on Jul 8, 2026

first commit for diffusion training

zirui
and
cursoragent
committed
docs(diffusion): unify single/multi-node torchrun launch in READMEs

zirui
and
cursoragent
committed
docs(diffusion): simplify unified run instructions
zirui
committed
fix(diffusion): broadcast Wan timesteps per sample

zirui
and
cursoragent
committed
Remove unused pip packages and clean up README.md
zirui
committed
fix launch command
zirui
committed
set aiter as default attention backend
zirui
committed
fix: add missing newline at end of examples/diffusion/README.md

Copilot
authored and
zirui
committed
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 and
zirui
committed
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
feat(diffusion): add FLUX training support

zirui
and
cursoragent
committed
fix(diffusion): make FLUX examples portable

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

zirui
and
cursoragent
committed
fix(diffusion): use AITER attention for FLUX

zirui
and
cursoragent
committed
fix(tests): add __init__.py to test dirs to prevent module name conflicts

Copilot
authored and
zirui
committed
fix(diffusion): address PR review comments

zirui
and
cursoragent
committed
docs(diffusion): generalize utils docstring

zirui
and
cursoragent
committed

Commits on Jul 15, 2026