From f2ac9ed6e3157ef347d04eeb238095ac4b0610d5 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 16:44:25 -0800 Subject: [PATCH] Update forge_loader.py --- modules_forge/forge_loader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/modules_forge/forge_loader.py b/modules_forge/forge_loader.py index 0c68c2cf..d2a2197e 100644 --- a/modules_forge/forge_loader.py +++ b/modules_forge/forge_loader.py @@ -241,17 +241,18 @@ def load_model_for_a1111(timer, checkpoint_info=None, state_dict=None): sd_model.decode_first_stage = patched_decode_first_stage sd_model.encode_first_stage = patched_encode_first_stage - sd_model.current_control_signals = { + sd_model.current_controlnet_signals = { 'input': [], 'middle': [], 'output': [] } + sd_model.current_controlnet_required_memory = 0 original_forward = sd_model.model.diffusion_model.forward def forge_unet_forward(*args, **kwargs): kwargs.update(dict( - control=sd_model.current_control_signals, + control=sd_model.current_controlnet_signals, transformer_options=sd_model.forge_objects.unet.model_options.get('transformer_options', {}) )) return original_forward(*args, **kwargs)