lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-11 10:29:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_sampler.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-28 09:15:48 -08:00
parent dafe28c025
commit 2ee5badc6f
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules_forge/forge_sampler.py
View File

@@ -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: