From 587c93ebff781632217895db1ef5f5e59e3cdc5e Mon Sep 17 00:00:00 2001 From: chaObserv <154517000+chaObserv@users.noreply.github.com> Date: Sun, 22 Sep 2024 18:27:13 +0800 Subject: [PATCH] Start lower_final_step in predictor earlier to stabilize --- comfy/k_diffusion/sampling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/k_diffusion/sampling.py b/comfy/k_diffusion/sampling.py index da2c0e0b..f4e39bfb 100644 --- a/comfy/k_diffusion/sampling.py +++ b/comfy/k_diffusion/sampling.py @@ -1096,7 +1096,7 @@ def sample_sa_solver(model, x, sigmas, extra_args=None, callback=None, disable=F sigma_prev_list.append(sigma) else: # Lower order final - predictor_order_used = min(predictor_order, i, len(sigmas) - i) + predictor_order_used = min(predictor_order, i, len(sigmas) - i - 1) corrector_order_used = min(corrector_order, i + 1, len(sigmas) - i + 1) tau_val = tau(sigma)