Jaret Burkett 4f91cb7148 Fix issue with gradient checkpointing and flux kontext

2025-06-26 19:03:12 -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 issue with gradient checkpointing and flux kontext

2025-06-26 19:03:12 -06:00

hidream

Fixed issue with training hidream when batch size is larger than 1

2025-04-21 17:26:29 +00:00

omnigen2

Work on mean flow. Minor bug fixes. Omnigen improvements

2025-06-26 13:46:20 -06:00

__init__.py

Added support for FLUX.1-Kontext-dev

2025-06-26 15:24:37 -06:00