298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix lambda_prev in corrector

Browse Source
This commit is contained in:
chaObserv 2024-09-17 23:21:52 +08:00
parent 92a630e737
commit 1782896502
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/k_diffusion/sa_solver.py
View File

@ -150,7 +150,7 @@ def adams_moulton_update_few_steps(order, x, tau, model_prev_list, sigma_prev_li
sigma_list = sigma_prev_list + [sigma] sigma_list = sigma_prev_list + [sigma]
lambda_list = [t_fn(sigma_list[-(i + 1)]) for i in range(order)] lambda_list = [t_fn(sigma_list[-(i + 1)]) for i in range(order)]
lambda_t = lambda_list[0] lambda_t = lambda_list[0]
lambda_prev = lambda_list[1] lambda_prev = lambda_list[1] if order >= 2 else t_fn(sigma_prev)
h = lambda_t - lambda_prev h = lambda_t - lambda_prev
gradient_coefficients = get_coefficients_fn(order, lambda_prev, lambda_t, lambda_list, tau) gradient_coefficients = get_coefficients_fn(order, lambda_prev, lambda_t, lambda_list, tau)