Start lower_final_step in predictor earlier to stabilize

This commit is contained in:
chaObserv 2024-09-22 18:27:13 +08:00
parent 1782896502
commit 587c93ebff

View File

@ -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)