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