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

Update unet_patcher.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-30 19:21:31 -08:00
parent aa5d960c8e
commit 50c82c06f2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/unet_patcher.py
View File

@@ -72,8 +72,8 @@ class UnetPatcher(ModelPatcher):
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):
for number in range(8):
for transformer_index in range(16):
self.set_model_patch_replace(patch, target, block_name, number, transformer_index)
return