Jaret Burkett a558d5b68f Move transformer back to device on aggresive wan 2.2 pipeline after generation.

2025-07-29 09:13:47 -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

wan22

Move transformer back to device on aggresive wan 2.2 pipeline after generation.

2025-07-29 09:13:47 -06:00

__init__.py

Add support for Wan2.2 5B

2025-07-29 05:31:54 -06:00