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

Update unet_patcher.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-30 11:48:57 -08:00
parent 8e6c36111e
commit 0f532823b2
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
modules_forge/unet_patcher.py
View File

@@ -34,12 +34,16 @@ class UnetPatcher(ModelPatcher):
pointer = pointer.previous_controlnet
return results
def add_conditioning_modifier(self, modifier, ensure_uniqueness=False):
if 'conditioning_modifiers' not in self.model_options:
self.model_options['conditioning_modifiers'] = []
def append_model_option(self, k, v, ensure_uniqueness=False):
if k not in self.model_options:
self.model_options[k] = []
if ensure_uniqueness and modifier in self.model_options['conditioning_modifiers']:
if ensure_uniqueness and v in self.model_options[k]:
return
self.model_options['conditioning_modifiers'].append(modifier)
self.model_options[k].append(v)
return
def add_conditioning_modifier(self, modifier, ensure_uniqueness=False):
self.append_model_option('conditioning_modifiers', modifier, ensure_uniqueness)
return