ai-toolkit/extensions_built_in/diffusion_models
Jaret Burkett 5e84bf0d0b Fixed issue with hidream-01 that could cause a weird nan state. Took forever to track down as it was 1 in 10 starts. 2026-05-28 12:48:24 -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
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 Fixed issue with hidream-01 that could cause a weird nan state. Took forever to track down as it was 1 in 10 starts. 2026-05-28 12:48:24 -06:00
ltx2 Fixed offload and quantize order of ltx 2.3 text encoder. 2026-04-07 15:11:50 -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
qwen_image Added decode latent method to qwen image model 2026-05-25 10:02:09 -06:00
wan22 Hotfix some issues with Wan models caused by diffusers and transformers updates 2026-04-16 20:53:50 +00:00
z_image Change z image divisibility 2026-05-28 09:09:36 -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 ZImage L2P 2026-05-22 14:58:28 -06:00