Jaret Burkett fbac1cb7f5 Dont force flash attention on hidream 01. Causes random issues and is slower.

2026-05-24 16:05:22 -06:00

..

src

Fix potential inconsistency with different attention mentods in hidream01

2026-05-13 09:08:11 -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