Tiny cleanup

2024-05-04 05:18:57 -06:00
parent e7c6ffe3c0
commit 985fc1f3e8
1 changed files with 1 additions and 1 deletions
@@ -469,7 +469,7 @@ class RestartPlan:
            restart_sigmas = calc_sigmas(
                restart_scheduler,
                n_restart,
-                max(model_sigma_min, sigmas[i + 1]),
+                max(model_sigma_min, s_min),
                s_max,
                model,
                device=device,