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).