lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-11 16:09:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

add to patcher

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 19:12:13 -08:00
parent 611b234792
commit 61b014f579
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/unet_patcher.py
View File

@@ -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']: