From 2ee5badc6f50ee9b40246b6622c561dcbb5a891c Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sun, 28 Jan 2024 09:15:48 -0800 Subject: [PATCH] Update forge_sampler.py --- modules_forge/forge_sampler.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/modules_forge/forge_sampler.py b/modules_forge/forge_sampler.py index 4e44aaac..010e24f3 100644 --- a/modules_forge/forge_sampler.py +++ b/modules_forge/forge_sampler.py @@ -67,8 +67,10 @@ def forge_sample(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]: - uncond[0]['model_conds']['c_concat'] = CONDRegular(image_cond_in) - cond[0]['model_conds']['c_concat'] = CONDRegular(image_cond_in) + for i in range(len(uncond)): + uncond[i]['model_conds']['c_concat'] = CONDRegular(image_cond_in) + for i in range(len(cond)): + cond[i]['model_conds']['c_concat'] = CONDRegular(image_cond_in) if control is not None: for h in cond + uncond: