ai-toolkit/extensions_built_in/diffusion_models
Jaret Burkett 611969ec1f Allow control image for omnigen training and sampling 2025-07-09 13:54:55 -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 Fixed issue with flux kontext forcing generation image sizes 2025-06-29 05:38:20 -06:00
hidream Fixed issue with training hidream when batch size is larger than 1 2025-04-21 17:26:29 +00:00
omnigen2 Allow control image for omnigen training and sampling 2025-07-09 13:54:55 -06:00
__init__.py Added support for FLUX.1-Kontext-dev 2025-06-26 15:24:37 -06:00