Jaret Burkett ffaf2f154a Fix issue with the way chroma handled gradient checkpointing.

2025-05-28 08:41:47 -06:00

..

__init__.py

Added support for Lodestone Rock's Chroma model

2025-04-05 13:21:36 -06:00

layers.py

Added support for Lodestone Rock's Chroma model

2025-04-05 13:21:36 -06:00

math.py

Allow fine tuning pruned versions of chroma. Allow flash attention 2 for chroma if it is installed.

2025-05-21 07:02:50 -06:00

model.py

Fix issue with the way chroma handled gradient checkpointing.

2025-05-28 08:41:47 -06:00