Commit f10b8948 authored by comfyanonymous's avatar comfyanonymous
Browse files

768-v support for uni_pc sampler.

parent ce0aeb10
......@@ -835,12 +835,16 @@ def sample_unipc(model, noise, image, sigmas, sampling_function, extra_args=None
device = noise.device
if model.inner_model.parameterization == "v":
model_type = "v"
else:
model_type = "noise"
model_fn = model_wrapper(
model.inner_model.apply_model,
sampling_function,
ns,
model_type="noise",
model_type=model_type,
guidance_type="uncond",
model_kwargs=extra_args,
)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment