Add ControlNet support.

This commit is contained in:
comfyanonymous
2023-02-16 10:38:08 -05:00
parent bc69fb5245
commit 4efa67fa12
9 changed files with 580 additions and 63 deletions

View File

@@ -856,13 +856,13 @@ def sample_unipc(model, noise, image, sigmas, sampling_function, extra_args=None
device = noise.device
if model.inner_model.parameterization == "v":
if model.parameterization == "v":
model_type = "v"
else:
model_type = "noise"
model_fn = model_wrapper(
model.inner_model.apply_model,
model.inner_model.inner_model.apply_model,
sampling_function,
ns,
model_type=model_type,