From 0d8eb4c5ba211ab468e270989a81c57c3783f465 Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Wed, 21 Aug 2024 11:01:59 -0700 Subject: [PATCH] fix #1375 --- backend/sampling/sampling_function.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/backend/sampling/sampling_function.py b/backend/sampling/sampling_function.py index 068177e3..9ae54d36 100644 --- a/backend/sampling/sampling_function.py +++ b/backend/sampling/sampling_function.py @@ -340,14 +340,18 @@ def sampling_function(self, denoiser_params, cond_scale, cond_composition): if image_cond_in.shape[0] == x.shape[0] \ and image_cond_in.shape[2] == x.shape[2] \ and image_cond_in.shape[3] == x.shape[3]: - for i in range(len(uncond)): - uncond[i]['model_conds']['c_concat'] = Condition(image_cond_in) + if uncond is not None: + for i in range(len(uncond)): + uncond[i]['model_conds']['c_concat'] = Condition(image_cond_in) for i in range(len(cond)): cond[i]['model_conds']['c_concat'] = Condition(image_cond_in) if control is not None: - for h in cond + uncond: + for h in cond: h['control'] = control + if uncond is not None: + for h in uncond: + h['control'] = control for modifier in model_options.get('conditioning_modifiers', []): model, x, timestep, uncond, cond, cond_scale, model_options, seed = modifier(model, x, timestep, uncond, cond, cond_scale, model_options, seed)