Raphael Walker 89e4ea0175 Add activations_shape info in UNet models (#7482) ...

* Add activations_shape info in UNet models

* activations_shape should be a list

2025-04-04 21:27:54 -04:00

..

diffusionmodules

Disable pytorch attention in VAE for AMD.

2025-02-14 05:42:14 -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 activations_shape info in UNet models (#7482)

2025-04-04 21:27:54 -04: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