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

Add FLUX Model - #99

Merged

suiyoubi merged 26 commits into

mainfrom

Feb 6, 2026

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