This repository was archived by the owner on May 20, 2026. It is now read-only.

Repository navigation

Add FLUX Model - #99

Merged

suiyoubi merged 26 commits into

mainfrom

aot/flux

Feb 6, 2026

Merged

Add FLUX Model#99
suiyoubi merged 26 commits into
mainfrom
aot/flux

Commits on Jan 8, 2026
- initial version with model fwd done
  suiyoubi
  committed
  7f4d682
Commits on Jan 20, 2026
- flux training
  suiyoubi
  committed
  5921a15
Commits on Jan 28, 2026
- add real dataset pipeline
  suiyoubi
  committed
  fc19ee9
Commits on Feb 3, 2026
- Remove unused utils.py file and clean up code in flux_pipeline.py and convert_checkpoints.py. Add functionality to save individual sample files in prepare_energon_dataset_flux.py.
  suiyoubi
  committed
  062faa4
- Fix dist ckpt issue
  suiyoubi
  committed
  a4b6c35
- simplify conversion script
  suiyoubi
  committed
  a4b5a0d
- ruff
  suiyoubi
  committed
  12c0f65
- lint
  suiyoubi
  committed
  b3fa5c7
- Merge branch 'main' of github.com:NVIDIA-NeMo/DFM into aot/flux
  suiyoubi
  committed
  743ed83
Commits on Feb 4, 2026
- imporve dataset preprocessing script for center cropping
  suiyoubi
  committed
  bd4683c
- unified flow matching pipeline
  suiyoubi
  committed
  bfb3507
- lint
  suiyoubi
  committed
  eec48f4
Commits on Feb 5, 2026
- repilicate id for tp sharding dist ckpt
  suiyoubi
  committed
  f57547e
- Add unit test
  suiyoubi
  committed
  148bcc1
- functional test
  suiyoubi
  committed
  9667238
- improve inf
  suiyoubi
  committed
  4b8e0cf
- fix tests
  suiyoubi
  committed
  2b2b28d
- add readme and rename inference pipeline
  suiyoubi
  committed
  7fe4a94
- comment improve
  suiyoubi
  committed
  6aad9c9
- restructure
  suiyoubi
  committed
  6f3e485
- select to run with either flux_step or flux_step_with_automodel
  suiyoubi
  committed
  b91d022
- trailing whitespace removed
  suiyoubi
  committed
  c748cca
- use timesteps from diffusers
  suiyoubi
  committed
  fd8bae2
- Add finetune script
  suiyoubi
  committed
  ff4f384
- doc fix
  suiyoubi
  committed
  7ea6788
Commits on Feb 6, 2026
- fix unit test
  suiyoubi
  committed
  d0f57eb

Commits

Commits on Jan 8, 2026

Commits on Jan 20, 2026

Commits on Jan 28, 2026

Commits on Feb 3, 2026

Commits on Feb 4, 2026

Commits on Feb 5, 2026

Commits on Feb 6, 2026