From b5ef71262f769ca8559475a9236e0faacdfaf68c Mon Sep 17 00:00:00 2001 From: blepping Date: Sun, 17 Mar 2024 13:17:25 -0600 Subject: [PATCH] Remove redundant seg assignment in restart wrapper --- restart_sampling.py | 1 - 1 file changed, 1 deletion(-) diff --git a/restart_sampling.py b/restart_sampling.py index 5845b04..771730c 100644 --- a/restart_sampling.py +++ b/restart_sampling.py @@ -208,7 +208,6 @@ class KSamplerRestartWrapper: seg = segments.get(s_min) if seg is None: continue - seg = segments[s_min] s_max, k, n_restart = seg['t_max'], seg['k'], seg['n'] seg_sigmas = calc_sigmas(self.restart_scheduler, n_restart, s_min, s_max, self.real_model, device=x.device)