Jaret Burkett 1c96b95617 Fix issue where sometimes the transformer does not get loaded properly.		2025-08-14 14:24:41 -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	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
omnigen2	Allow control image for omnigen training and sampling	2025-07-09 13:54:55 -06:00
qwen_image	Added a flushg during sampling to prevent spikes on low vram qwen	2025-08-12 12:57:18 -06:00
wan22	Fix issue where sometimes the transformer does not get loaded properly.	2025-08-14 14:24:41 -06:00
__init__.py	Wan22 14b training is working, still need tons of testing and some bug fixes	2025-08-14 13:03:27 -06:00