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