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

Fix uni_pc sampler not working with 1 or 2 steps.

Browse Source
This commit is contained in:
comfyanonymous 2023-02-26 04:01:01 -05:00
parent 8f75047d9c
commit 3cd7d84b53
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/extra_samplers/uni_pc.py
View File

@ -875,8 +875,9 @@ def sample_unipc(model, noise, image, sigmas, sampling_function, max_denoise, ex
model_kwargs=extra_args,
)
order = min(3, len(timesteps) - 1)
uni_pc = UniPC(model_fn, ns, predict_x0=True, thresholding=False, noise_mask=noise_mask, masked_image=image, noise=noise, variant=variant)
x = uni_pc.sample(img, timesteps=timesteps, skip_type="time_uniform", method="multistep", order=3, lower_order_final=True)
x = uni_pc.sample(img, timesteps=timesteps, skip_type="time_uniform", method="multistep", order=order, lower_order_final=True)
if not to_zero:
x /= ns.marginal_alpha(timesteps[-1])
return x