From 7e0c454dca92c9000faf34391012ea5890cb44de Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 27 Jan 2024 16:46:22 -0800 Subject: [PATCH] Update sd_forge_controlnet_example.py --- .../scripts/sd_forge_controlnet_example.py | 2 ++ 1 file changed, 2 insertions(+) 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 fea77c50..4b167cda 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 @@ -53,7 +53,9 @@ class ControlNetExampleForge(scripts.Script): controlnet = load_controlnet(controlnet_canny_path) 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) return