Function normalization

ldm/modules/diffusionmodules/util.py:199–205

Make a standard normalization layer. :param channels: number of input channels. :return: an nn.Module for normalization.

(channels, norm_channel=32)

def normalization(channels, norm_channel=32):
    """
    Make a standard normalization layer.
    :param channels: number of input channels.
    :return: an nn.Module for normalization.
    """
    return GroupNorm32(norm_channel, channels)


# PyTorch 1.7 has SiLU, but we support PyTorch 1.5.