tiny_diff.models.vae.layers

Classes

LatentAlignLayer(in_channels, out_channels)

Hidden layer that proyect into the latent space.