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

Update sd_forge_controlnet_example.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-27 19:40:58 -08:00
parent 1b4f2a4411
commit 38e281a50a
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions-builtin/sd_forge_controlnet_example/scripts/sd_forge_controlnet_example.py
View File

@@ -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