ai-toolkit/extensions_built_in/diffusion_models/flux2/src
Jaret Burkett e82cf6eec2 Fixed issue that prevented full fine-tuning of flux2 models when using gradient checkpointing 2026-02-06 16:18:43 -07:00
..
__init__.py Add initial support for FLUX.2 2025-11-18 11:17:38 -07:00
autoencoder.py Add initial support for FLUX.2 2025-11-18 11:17:38 -07: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