298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

768-v support for uni_pc sampler.

Browse Source
This commit is contained in:
comfyanonymous 2023-02-11 04:34:58 -05:00
parent ce0aeb109e
commit f10b8948c3
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/extra_samplers/uni_pc.py
View File

@ -835,12 +835,16 @@ def sample_unipc(model, noise, image, sigmas, sampling_function, extra_args=None
device = noise.device device = noise.device
if model.inner_model.parameterization == "v":
model_type = "v"
else:
model_type = "noise"
model_fn = model_wrapper( model_fn = model_wrapper(
model.inner_model.apply_model, model.inner_model.apply_model,
sampling_function, sampling_function,
ns, ns,
model_type="noise", model_type=model_type,
guidance_type="uncond", guidance_type="uncond",
model_kwargs=extra_args, model_kwargs=extra_args,
) )