diffusion\_models.models.vae ============================ .. automodule:: diffusion_models.models.vae .. rubric:: Classes .. autosummary:: :toctree: :template: custom-class-template.rst ResNet18Decoder ResNet18Encoder ResNetBlock ResNetDecoderBlock SkipDownSampler SkipUpSampler VariationalAutoencoder