tiny_diff.models.vae.components

Classes

ConvDecoder([final_act])

ConvDecoder for a VAE model.

ConvEncoder([z_channels, attn_channels, ...])

ConvEncoder for a VAE model.

ConvVAEComponent([z_channels, ...])

Class representing a VAE model component (encoder/decoder).