ai-toolkit/extensions_built_in/diffusion_models
Jaret Burkett 57d407cfd4 Add support for training lodestones/Zeta-Chroma 2026-03-01 12:52:29 -07:00
..
chroma Fix issue with chroma when not quantizing 2025-10-19 12:13:05 -06:00
f_light Added initial support for f-lite model 2025-05-01 11:15:18 -06:00
flux2 Fixed issue that prevented full fine-tuning of flux2 models when using gradient checkpointing 2026-02-06 16:18:43 -07:00
flux_kontext fix: Guidance incorrect shape 2025-07-18 12:49:18 +02:00
hidream Added initial support for Hidream E1 training 2025-07-27 15:12:56 -06:00
ltx2 Shrink text embeds to max token length for LTX-2. Drastically reduces cached text embedding sizes 2026-01-28 12:54:49 -07:00
omnigen2 Allow control image for omnigen training and sampling 2025-07-09 13:54:55 -06:00
qwen_image Fixed issue where the control images would sometimes be ignored on qwen_image_edit_2511 2025-12-30 16:40:49 +00:00
wan22 Added initial support for layer offloading wit Wan 2.2 14B models. 2025-10-20 14:54:30 -06:00
z_image Fix issue that prevented ramtorch layer offloading with z_image 2025-12-02 16:14:34 -07:00
zeta_chroma Add support for training lodestones/Zeta-Chroma 2026-03-01 12:52:29 -07:00
__init__.py Add support for training lodestones/Zeta-Chroma 2026-03-01 12:52:29 -07:00