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