Loading modules/sd_samplers_kdiffusion.py +4 −1 Original line number Diff line number Diff line Loading @@ -67,7 +67,10 @@ def restart_sampler(model, x, sigmas, extra_args=None, callback=None, disable=No if restart_list is None: if steps >= 20: restart_steps = 9 restart_times = 2 if steps >= 36 else 1 restart_times = 1 if steps >= 36: restart_steps = steps // 4 restart_times = 2 sigmas = get_sigmas_karras(steps - restart_steps * restart_times, sigmas[-2].item(), sigmas[0].item(), device=sigmas.device) restart_list = {0.1: [restart_steps + 1, restart_times, 2]} else: Loading Loading
modules/sd_samplers_kdiffusion.py +4 −1 Original line number Diff line number Diff line Loading @@ -67,7 +67,10 @@ def restart_sampler(model, x, sigmas, extra_args=None, callback=None, disable=No if restart_list is None: if steps >= 20: restart_steps = 9 restart_times = 2 if steps >= 36 else 1 restart_times = 1 if steps >= 36: restart_steps = steps // 4 restart_times = 2 sigmas = get_sigmas_karras(steps - restart_steps * restart_times, sigmas[-2].item(), sigmas[0].item(), device=sigmas.device) restart_list = {0.1: [restart_steps + 1, restart_times, 2]} else: Loading
