From b592142f3b46852263747a4efd0d244ad17b5bb3 Mon Sep 17 00:00:00 2001 From: licyk <76895225+licyk@users.noreply.github.com> Date: Mon, 28 Oct 2024 20:15:37 +0800 Subject: [PATCH] fix multidiffusion integrated with controlnet (#2199) --- .../lib_multidiffusion/tiled_diffusion.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py b/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py index ef8f2a4b..c11bafe3 100644 --- a/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py +++ b/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py @@ -578,9 +578,8 @@ class MixtureOfDiffusers(AbstractDiffusion): # controlnet # self.switch_controlnet_tensors(batch_id, N, len(bboxes), is_denoise=True) if 'control' in c_in: - control = c_in['control'] self.process_controlnet(x_tile.shape, x_tile.dtype, c_in, cond_or_uncond, bboxes, N, batch_id) - c_tile['control'] = control.get_control(x_tile, t_tile, c_tile, len(cond_or_uncond)) + c_tile['control'] = c_in['control_model'].get_control(x_tile, t_tile, c_tile, len(cond_or_uncond)) # stablesr # self.switch_stablesr_tensors(batch_id)