Jaret Burkett 5e84bf0d0b 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
..
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	Dont force flash attention on hidream 01. Causes random issues and is slower.	2026-05-24 16:05:22 -06:00