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

lin

Browse Source
This commit is contained in:
lllyasviel
2024-02-07 20:14:02 -08:00
parent a1670c536d
commit 49ec325f6a
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
ldm_patched/modules/controlnet.py
View File

@@ -67,7 +67,7 @@ def compute_controlnet_weighting(control, cnet):
if not isinstance(control_signal, torch.Tensor):
continue
B, C, H, W = control_signal.shape
positive_weight = 1.0
@@ -258,7 +258,8 @@ class ControlNet(ControlBase):
controlnet_model_function_wrapper = to.get('controlnet_model_function_wrapper', None)
if controlnet_model_function_wrapper is not None:
wrapper_args = dict(x=x_noisy.to(dtype), hint=self.cond_hint, timesteps=timestep.float(), context=context.to(dtype), y=y)
wrapper_args = dict(x=x_noisy.to(dtype), hint=self.cond_hint, timesteps=timestep.float(),
context=context.to(dtype), y=y)
wrapper_args['model'] = self
wrapper_args['inner_model'] = self.control_model
control = controlnet_model_function_wrapper(**wrapper_args)