Jedrzej Kosinski 51a30c2ad7 Make sure wrap_attn doesn't make itself recurse infinitely, attempt to load SageAttention and FlashAttention if not enabled so that they can be marked as available or not, create registry for available attention		2025-08-28 18:53:20 -07:00
..
diffusionmodules	Add elementwise fusions (#9495)	2025-08-22 19:39:15 -04:00
distributions	Small optimizations.	2024-12-18 18:23:28 -05:00
encoders	Make unclip more deterministic.	2024-01-14 17:28:31 -05:00
attention.py	Make sure wrap_attn doesn't make itself recurse infinitely, attempt to load SageAttention and FlashAttention if not enabled so that they can be marked as available or not, create registry for available attention	2025-08-28 18:53:20 -07:00
ema.py	Initial commit.	2023-01-16 22:37:14 -05:00
sub_quadratic_attention.py	Fix pytorch warning. (#8593)	2025-06-19 11:04:52 -04:00
temporal_ae.py	Basic Hunyuan Video model support.	2024-12-16 19:35:40 -05:00