Jaret Burkett 5cb54ba9cc Allow setting weight saving flag on hidream_o1

2026-07-16 07:40:25 -06:00

..

src

Fixed issue with hidream-01 that could cause a weird nan state. Took forever to track down as it was 1 in 10 starts.

2026-05-28 12:48:24 -06:00

__init__.py

Add support for fine-tuning Hidream O1 (#831)

2026-05-12 11:15:16 -06:00

hidream_e1_model.py

Added initial support for Hidream E1 training

2025-07-27 15:12:56 -06:00

hidream_model.py

Added initial support for Hidream E1 training

2025-07-27 15:12:56 -06:00

hidream_o1_model.py

Allow setting weight saving flag on hidream_o1

2026-07-16 07:40:25 -06:00