Jaret Burkett add83df5cc Fixed issue with training hidream when batch size is larger than 1

2025-04-21 17:26:29 +00:00

..

src

Make flash attn optional. Handle larger batch sizes.

2025-04-14 14:34:46 +00:00

__init__.py

Initial support for hidream. Still a WIP

2025-04-13 13:50:11 -06:00

hidream_model.py

Fixed issue with training hidream when batch size is larger than 1

2025-04-21 17:26:29 +00:00