tiny_diff.models.unet.layers

Classes

DownUNetLayer(kernel_size, **kwargs)	UNet layer that performs downsampling.
TimeEmbedding(channels)	Embedding layer for time in a diffusion process.
UNetLayer(in_channels, out_channels[, n_blocks])	UNet's first half layer with residual layers + embeddings.
UNetLayerMixin()	Mixin for UNet's layers without skip connections.
UNetSkipLayer(skip_in_channels, ...)	UNet's second half layer with residual layers + embeddings + skip connections.
UNetSkipLayerMixin()	Mixin for UNet's layers with skip connections.
UpUNetLayer(skip_in_channels, ...)	UNet layer that performs upsampling.