From 5d3071331a479389cd0f28b7bbd10ca7370b8011 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 27 Jan 2024 16:48:04 -0800 Subject: [PATCH] Update sd_forge_controlnet_example.py --- .../scripts/sd_forge_controlnet_example.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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 4b167cda..be695ac1 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 @@ -52,11 +52,15 @@ class ControlNetExampleForge(scripts.Script): print('The model [control_v11p_sd15_canny_fp16.safetensors] download finished.') controlnet = load_controlnet(controlnet_canny_path) + print('Controlnet loaded.') + input_image = cv2.resize(input_image, (p.height, p.width)) canny_image = cv2.Canny(input_image, 100, 200) # Display preprocessor result. Called every batch. Cache in your own way. - p.extra_result_images.append(input_image) + p.extra_result_images.append(canny_image) + + print('Preprocessor Canny finished.') return