From 985fc1f3e80d810ef75211d7d29ed5429a1928cb Mon Sep 17 00:00:00 2001
From: blepping <blepping@tuta.io>
Date: Sat, 4 May 2024 05:18:57 -0600
Subject: [PATCH] Tiny cleanup

---
 restart_sampling.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/restart_sampling.py b/restart_sampling.py
index a296f12..d2b972a 100644
--- a/restart_sampling.py
+++ b/restart_sampling.py
@@ -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,