lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-27 09:41:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-28 08:14:59 -08:00
parent 79e462e746
commit b066e1b9f6
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules_forge/controlnet.py
View File

@@ -72,6 +72,7 @@ def compute_controlnet_weighting(
return control return control
cond_or_uncond = transformer_options['cond_or_uncond'] cond_or_uncond = transformer_options['cond_or_uncond']
cond_or_uncond_size = transformer_options['cond_or_uncond_size']
sigmas = transformer_options['sigmas'] sigmas = transformer_options['sigmas']
if advanced_sigma_weighting is not None: if advanced_sigma_weighting is not None:

2
modules_forge/patch_basic.py
View File

@@ -129,6 +129,7 @@ def patched_calc_cond_uncond_batch(model, cond, uncond, x_in, timestep, model_op
patches = p.patches patches = p.patches
batch_chunks = len(cond_or_uncond) batch_chunks = len(cond_or_uncond)
cond_or_uncond_size = int(timestep.shape[0])
input_x = torch.cat(input_x) input_x = torch.cat(input_x)
c = cond_cat(c) c = cond_cat(c)
timestep_ = torch.cat([timestep] * batch_chunks) timestep_ = torch.cat([timestep] * batch_chunks)
@@ -150,6 +151,7 @@ def patched_calc_cond_uncond_batch(model, cond, uncond, x_in, timestep, model_op
transformer_options["cond_or_uncond"] = cond_or_uncond[:] transformer_options["cond_or_uncond"] = cond_or_uncond[:]
transformer_options["sigmas"] = timestep transformer_options["sigmas"] = timestep
transformer_options["cond_or_uncond_size"] = cond_or_uncond_size
c['transformer_options'] = transformer_options c['transformer_options'] = transformer_options