ai-toolkit/extensions_built_in/diffusion_models
Jaret Burkett a4bbe167ce Added reference token attention isolation (kv_cache) for Krea2 edit training. Same training cost with significant inference speed up. 2x inference speedup. 2026-07-09 12:00:55 -06:00
..
boogu_image Add support for Boogu Image and Boogu Image Edit 2026-06-18 15:05:49 -06:00
chroma Fix issue with chroma when not quantizing 2025-10-19 12:13:05 -06:00
ernie_image Fix ernie unpatchify 2026-04-17 12:03:39 -06:00
example_model Added additional information on addine new models and some additional gotchas 2026-06-18 15:04:50 -06:00
f_light Added initial support for f-lite model 2025-05-01 11:15:18 -06:00
flux2 Made it possible to use the flux 2 small decoder VAE when setting the vae_path manually for flux2 models 2026-04-30 05:42:31 -06:00
flux_kontext fix: Guidance incorrect shape 2025-07-18 12:49:18 +02:00
hidream Rework merge_network_on_save to handle dequantization on merging and saving for much more efficient full finetuning. 2026-06-25 13:19:38 -06:00
ideogram4 Apply ideogram dynamic shifting to sampling 2026-06-16 18:52:16 -06:00
krea2 Added reference token attention isolation (kv_cache) for Krea2 edit training. Same training cost with significant inference speed up. 2x inference speedup. 2026-07-09 12:00:55 -06:00
ltx2 Apply a loss mask for first frame conditioning for wan and ltx 2026-07-07 09:05:53 -06:00
nucleus_image Add support for Nucleus-Image 2026-04-16 13:09:10 -06:00
omnigen2 Allow control image for omnigen training and sampling 2025-07-09 13:54:55 -06:00
prx_pixel_t2i Add support for PRX Pixel T2I 2026-06-13 09:47:53 -06:00
qwen_image Fix breaking change with diffusers qwen image 2026-06-26 10:04:35 -06:00
wan22 Apply a loss mask for first frame conditioning for wan and ltx 2026-07-07 09:05:53 -06:00
z_image Fix sampling issue on zimage turbo and krea2 turbo models 2026-07-08 12:34:39 -06:00
zeta_chroma Added some experimental low step things for zeta 2026-04-13 09:37:34 -06:00
__init__.py Add support for Krea2 (#906) 2026-06-23 09:18:14 -06:00