comfyanonymous 129d8908f7 Add argument to skip the output reshaping in the attention functions.

2025-01-10 06:27:37 -05:00

..

diffusionmodules

Replace print with logging (#6138)

2024-12-20 16:24:55 -05:00

distributions

Small optimizations.

2024-12-18 18:23:28 -05:00

encoders

…

attention.py

Add argument to skip the output reshaping in the attention functions.

2025-01-10 06:27:37 -05:00

ema.py

…

sub_quadratic_attention.py

Fix and enforce all ruff W rules.

2025-01-01 03:08:33 -05:00

temporal_ae.py

…