From 32fd7e7e5d8a96e9da85f5f142846eeba0457cc4 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Mon, 29 Jan 2024 18:24:09 -0800 Subject: [PATCH] Update controlnet.py --- .../sd_forge_controlnet/scripts/controlnet.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py index 1bd9562f..eb8dd6a8 100644 --- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py +++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py @@ -456,13 +456,12 @@ class ControlNetForForgeOfficial(scripts.Script): params.control_cond_for_hr_fix = preprocessor_output p.extra_result_images.append(input_image) + model_filename = global_state.get_controlnet_filename(unit.model) + + params.model = cached_controlnet_loader(model_filename) params.preprocessor = preprocessor - model_filename = global_state.get_controlnet_filename(unit.model) - params.model = cached_controlnet_loader(model_filename) - logger.info(f"Current ControlNet: {model_filename}") - return def process_unit_before_every_sampling(self,