Repository navigation

Enable flux multinode - #1

Closed

cpersson-amd wants to merge 19 commits into

jianhan/synthetic_data_iteratorfrom

cpersson/flux_multinode

Commits on Jan 20, 2026
Commits on Jan 21, 2026
Commits on Jan 22, 2026
- fp8 bug for batch_size setting error (#317)
  susanbao
  authored
  a385c8e
Commits on Jan 23, 2026
- Add batch divisibility check for VAE input sharding (#316)
  prishajain1
  authored
  5b0b8cf
Commits on Jan 24, 2026
- Add GitHub Action to add "pull ready" to a PR upon successful checks (#319)
  michelle-yooh
  authored
  ad56886
Commits on Jan 29, 2026
- Enable JIT Compilation of WAN VAE Encoder/Decoder Forward Passes (#320)
  prishajain1
  authored
  63b5ed7
- Add LoRA support for WAN models (#308)
  Perseus14
  authored
  9622341
- Move the top level 'tests/' into `src/maxdiffusion/tests/` as a legacy (#322)
  michelle-yooh
  authored
  1e1058a
- feat: add general synthetic data iterator and examples for WAN and FLUX.
  
  meijianhan
  authored and
  jianhan-amd
  committed
  54a7518
- doc: update README.md with synthetic data iterator usage.
  
  meijianhan
  authored and
  jianhan-amd
  committed
  206f7db
- feat: add loss_scaling_factor for matching the flux loss.
  jianhan-amd
  committed
  4793049
- feat: add multi-node running scripts. Add dockerfile with IB driver installation.
  jianhan-amd
  committed
  26aeeab
Commits on Jan 30, 2026
- feat: add multi-node scripts without using Slurm.
  root
  committed
  0bdcd72
Commits on Jan 31, 2026
- fix: fix the WAN1.3B model loading bug
  root
  committed
  44a148b
Commits on Feb 5, 2026
- enable flux multinode training
  cpersson-amd
  committed
  16bedad