TinyDiff
v0.1.0-3-g301651b
Usage
API
tiny_diff
tiny_diff.constants
tiny_diff.losses
tiny_diff.models
tiny_diff.models.conv_model
tiny_diff.models.discriminator
tiny_diff.models.unet
tiny_diff.models.vae
tiny_diff.modules
tiny_diff.schedulers
tiny_diff.utils
TinyDiff
API
tiny_diff
tiny_diff.models
tiny_diff.models.unet
tiny_diff.models.unet.model
View page source
tiny_diff.models.unet.model
Classes
UNet
([time_channels, attn_at_layer, ...])
UNet model used for diffusion tasks.