lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-27 03:19:47 +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

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
batch_chunks = len(cond_or_uncond)
cond_or_uncond_size = int(timestep.shape[0])
input_x = torch.cat(input_x)
c = cond_cat(c)
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["sigmas"] = timestep
transformer_options["cond_or_uncond_size"] = cond_or_uncond_size
c['transformer_options'] = transformer_options