Function spatial_norm_thresholding

ldm/models/diffusion/sampling_util.py:19–22

(x0, value)

def spatial_norm_thresholding(x0, value):
    # b c h w
    s = x0.pow(2).mean(1, keepdim=True).sqrt().clamp(min=value)
    return x0 * (value / s)