Function norm_thresholding

ldm/models/diffusion/sampling_util.py:14–16

(x0, value)

def norm_thresholding(x0, value):
    s = append_dims(x0.pow(2).flatten(1).mean(1).sqrt().clamp(min=value), x0.ndim)
    return x0 * (value / s)


def spatial_norm_thresholding(x0, value):