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