UNet2DModel: 1039, AttnDownBlock2D: 1033, AttnUpBlock2D: 1033, AttnSkipDownBlock2D: 6, AttnSkipUpBlock2D: 6
VQModel: 13, UNetMidBlock2D: 13, AttnDownEncoderBlock2D: 7, AttnUpDecoderBlock2D: 7
AutoencoderKL: 2927, UNetMidBlock2D: 2927, AttnDownEncoderBlock2D: 1, AttnUpDecoderBlock2D: 1