comfyanonymous 008761166f Optimize first attention block in cosmos VAE.

2025-01-15 21:48:46 -05:00

..

diffusionmodules

Optimize first attention block in cosmos VAE.

2025-01-15 21:48:46 -05: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

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

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

ema.py

Initial commit.

2023-01-16 22:37:14 -05:00

sub_quadratic_attention.py

Fix and enforce all ruff W rules.

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

temporal_ae.py

Basic Hunyuan Video model support.

2024-12-16 19:35:40 -05:00