Jaret Burkett 7ccec8ec2c Add checkpointing and a proper decode for flux 2 VAEs so they can be used with DFE

2026-04-30 04:27:13 -06:00

..

__init__.py

Add initial support for FLUX.2

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

autoencoder.py

Add checkpointing and a proper decode for flux 2 VAEs so they can be used with DFE

2026-04-30 04:27:13 -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