From 38e281a50aed5f142a1eb046bf3c70fbbfbf6636 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 27 Jan 2024 19:40:58 -0800 Subject: [PATCH] Update sd_forge_controlnet_example.py --- .../scripts/sd_forge_controlnet_example.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py b/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py index 3d66a9b3..5f7f7395 100644 --- a/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py +++ b/extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py @@ -12,6 +12,8 @@ from ldm_patched.modules.controlnet import load_controlnet class ControlNetExampleForge(scripts.Script): + model = None + def title(self): return "ControlNet Example for Developers" @@ -47,7 +49,7 @@ class ControlNetExampleForge(scripts.Script): ) print('The model [control_v11p_sd15_canny_fp16.safetensors] download finished.') - controlnet = load_controlnet(controlnet_canny_path) + self.model = load_controlnet(controlnet_canny_path) print('Controlnet loaded.') return @@ -58,7 +60,7 @@ class ControlNetExampleForge(scripts.Script): input_image, funny_slider = script_args - if input_image is None: + if input_image is None or self.model is None: return B, C, H, W = kwargs['noise'].shape # latent_shape