Jaret Burkett deb409085a Made it possible to use the flux 2 small decoder VAE when setting the vae_path manually for flux2 models

2026-04-30 05:42:31 -06:00

..

__init__.py

Add initial support for FLUX.2

2025-11-18 11:17:38 -07:00

autoencoder.py

Made it possible to use the flux 2 small decoder VAE when setting the vae_path manually for flux2 models

2026-04-30 05:42:31 -06:00

model.py

Fixed issue that prevented full fine-tuning of flux2 models when using gradient checkpointing

2026-02-06 16:18:43 -07:00

pipeline.py

Add support for FLUX.2 klein base models

2026-01-17 17:46:25 -07:00

sampling.py

Add initial support for FLUX.2

2025-11-18 11:17:38 -07:00