lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-30 04:49:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update patch_basic.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-28 07:47:41 -08:00
parent d42cfc2a9f
commit 8158e31d80
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/patch_basic.py
View File

@@ -130,7 +130,6 @@ def patched_calc_cond_uncond_batch(model, cond, uncond, x_in, timestep, model_op
timestep_ = torch.cat([timestep] * batch_chunks)
if control is not None:
control.current_cond_or_uncond = cond_or_uncond
c['control'] = control.get_control(input_x, timestep_, c, len(cond_or_uncond))
transformer_options = {}
@@ -153,6 +152,9 @@ def patched_calc_cond_uncond_batch(model, cond, uncond, x_in, timestep, model_op
c['transformer_options'] = transformer_options
if control is not None:
control.transformer_options = transformer_options
if 'model_function_wrapper' in model_options:
output = model_options['model_function_wrapper'](model.apply_model, {"input": input_x, "timestep": timestep_, "c": c, "cond_or_uncond": cond_or_uncond}).chunk(batch_chunks)
else: