From 424b225fb30dc9f873c5d37bcf424cc120979428 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 27 Jan 2024 22:31:12 -0800 Subject: [PATCH] Update sd_forge_controlnet_example.py --- .../scripts/sd_forge_controlnet_example.py | 2 +- 1 file changed, 1 insertion(+), 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 1c56ca37..2277ae53 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 @@ -70,7 +70,7 @@ class ControlNetExampleForge(scripts.Script): width = W * 8 input_image = cv2.resize(input_image, (height, width)) - canny_image = cv2.Canny(input_image, 100, 200) + canny_image = cv2.cvtColor(cv2.Canny(input_image, 100, 200), cv2.COLOR_GRAY2RGB) # Output preprocessor result. Now called every sampling. Cache in your own way. p.extra_result_images.append(canny_image)