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

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-30 19:15:47 -08:00
parent df673fa6dc
commit aa5d960c8e
2 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules_forge/unet_patcher.py
View File

@@ -70,6 +70,13 @@ class UnetPatcher(ModelPatcher):
self.append_transformer_option('block_modifiers', modifier, ensure_uniqueness)
return
def set_model_replace_all(self, patch, target="attn1"):
for block_name in ['input', 'middle', 'output']:
for number in range(64):
for transformer_index in range(64):
self.set_model_patch_replace(patch, target, block_name, number, transformer_index)
return
def forge_unet_forward(self, x, timesteps=None, context=None, y=None, control=None, transformer_options=None, **kwargs):
if transformer_options is None: