ai-toolkit/extensions_built_in/diffusion_models
Jaret Burkett 60ef2f1df7 Added support for FLUX.1-Kontext-dev 2025-06-26 15:24:37 -06:00
..
chroma Fix issue with the way chroma handled gradient checkpointing. 2025-05-28 08:41:47 -06:00
f_light Added initial support for f-lite model 2025-05-01 11:15:18 -06:00
flux_kontext Added support for FLUX.1-Kontext-dev 2025-06-26 15:24:37 -06:00
hidream Fixed issue with training hidream when batch size is larger than 1 2025-04-21 17:26:29 +00:00
omnigen2 Work on mean flow. Minor bug fixes. Omnigen improvements 2025-06-26 13:46:20 -06:00
__init__.py Added support for FLUX.1-Kontext-dev 2025-06-26 15:24:37 -06:00