deeplay.components.diffusion.attention_unet Module

Classes

AttentionBlock(*args, **kwargs)	Applies attention mechanism to the input tensor.
AttentionUNet(*args, **kwargs)	Attention UNet.
Block(*args, **kwargs)	Base block of the attention UNet.
DeeplayModule(*args, **kwargs)	A base class for creating configurable, extensible modules with dynamic initialization and argument management.
FeatureIntegrationModule(*args, **kwargs)	Integrates the time and context information to the feature maps through residual connections and attention mechanisms.
Layer(*args, **kwargs)
LayerActivationNormalization(*args, **kwargs)
LayerList(*args, **kwargs)
Sequential(*args, **kwargs)
UNetDecoder(*args, **kwargs)	UNet decoder.
UNetEncoder(*args, **kwargs)	UNet encoder.