Function ema_update_dict

k_diffusion/utils.py:452–459

(values, updates, decay)

def ema_update_dict(values, updates, decay):
    for k, v in updates.items():
        if k not in values:
            values[k] = v
        else:
            values[k] *= decay
            values[k] += (1 - decay) * v
    return values


def pyramid_noise_like(noise, iterations=6, discount=0.3):