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
distributions
encoders
attention.py	Add activations_shape info in UNet models (#7482)	2025-04-04 21:27:54 -04:00
ema.py
sub_quadratic_attention.py
temporal_ae.py