Start lower_final_step in predictor earlier to stabilize

comfyanonymous · Sep 22, 2024 · 587c93e · 587c93e
1 parent 1782896
commit 587c93e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)