Function get_cogvideo_dynamic_cfg_scale_list

trainer/scheduler.py:130–139

(cfg_scale, num_inference_steps,
                                        timesteps)

def get_cogvideo_dynamic_cfg_scale_list(cfg_scale, num_inference_steps,
                                        timesteps):

    cfg_list = [
        1 + cfg_scale * ((1 - math.cos(math.pi * (
            (num_inference_steps - t.item()) / num_inference_steps)**5.0)) / 2)
        for t in timesteps
    ]

    return cfg_list


# https://arxiv.org/pdf/2404.07724 Applying Guidance in a Limited Interval Improves Sample and Distribution Quality in Diffusion Models