Jedrzej Kosinski b58db6934c Looking into a @wrap_attn decorator to look for 'optimized_attention_override' entry in transformer_options

2025-08-27 14:18:18 -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

Looking into a @wrap_attn decorator to look for 'optimized_attention_override' entry in transformer_options

2025-08-27 14:18:18 -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