Block

class deeplay.components.diffusion.attention_unet.Block(*args, **kwargs)

Bases: LayerActivationNormalization

Base block of the attention UNet. It consists of a convolutional layer, a group normalization layer, and a GELU activation layer.