From 61b014f5794e3990c79d37205998a3e5b3d6f855 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 31 Jan 2024 19:12:13 -0800 Subject: [PATCH] add to patcher --- modules_forge/unet_patcher.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules_forge/unet_patcher.py b/modules_forge/unet_patcher.py index 1d4b3122..c3c655d8 100644 --- a/modules_forge/unet_patcher.py +++ b/modules_forge/unet_patcher.py @@ -67,6 +67,10 @@ class UnetPatcher(ModelPatcher): def add_block_modifier(self, modifier, ensure_uniqueness=False): self.append_transformer_option('block_modifiers', modifier, ensure_uniqueness) return + + def add_controlnet_conditioning_modifier(self, modifier, ensure_uniqueness=False): + self.append_transformer_option('controlnet_conditioning_modifiers', modifier, ensure_uniqueness) + return def set_model_replace_all(self, patch, target="attn1"): for block_name in ['input', 'middle', 'output']: