From 2074e13253a9a9f6509e1a1fee4b8ca926394acd Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 27 Jan 2024 19:42:48 -0800 Subject: [PATCH] Update sd_forge_controlnet_example.py --- .../scripts/sd_forge_controlnet_example.py | 12 ++++++++++++ 1 file changed, 12 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 5f7f7395..00947f93 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 @@ -75,6 +75,18 @@ class ControlNetExampleForge(scripts.Script): print('Preprocessor Canny finished.') + unet = p.sd_model.forge_objects.unet + + # unet = opFreeU_V2.patch(unet, freeu_b1, freeu_b2, freeu_s1, freeu_s2)[0] + + p.sd_model.forge_objects.unet = unet + + # Below codes will add some logs to the texts below the image outputs on UI. + # The extra_generation_params does not influence results. + p.extra_generation_params.update(dict( + controlnet_info='You should see these texts below output images!', + )) + return