new restart scheme

modules/sd_samplers_kdiffusion.py

@@ -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: