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

Allow controlnet_model_function_wrapper

Browse Source 
for animatediff to manage controlnet batch slicing window
This commit is contained in:
lllyasviel
2024-02-07 20:06:23 -08:00
parent 383aaca1eb
commit a1670c536d
2 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
modules_forge/unet_patcher.py
View File

@@ -98,6 +98,13 @@ class UnetPatcher(ModelPatcher):
to[k].append(v)
return
def set_transformer_option(self, k, v):
if 'transformer_options' not in self.model_options:
self.model_options['transformer_options'] = {}
self.model_options['transformer_options'][k] = v
return
def add_conditioning_modifier(self, modifier, ensure_uniqueness=False):
self.append_model_option('conditioning_modifiers', modifier, ensure_uniqueness)
return
@@ -150,6 +157,10 @@ class UnetPatcher(ModelPatcher):
self.append_transformer_option('controlnet_conditioning_modifiers', modifier, ensure_uniqueness)
return
def set_controlnet_model_function_wrapper(self, wrapper):
self.set_transformer_option('controlnet_model_function_wrapper', wrapper)
return
def set_model_replace_all(self, patch, target="attn1"):
for block_name in ['input', 'middle', 'output']:
for number in range(16):