Update unet_patcher.py

This commit is contained in:
lllyasviel
2024-02-01 11:55:08 -08:00
parent 412bc10a01
commit 988960ae37

View File

@@ -92,7 +92,7 @@ class UnetPatcher(ModelPatcher):
def set_model_replace_all(self, patch, target="attn1"):
for block_name in ['input', 'middle', 'output']:
for number in range(8):
for number in range(16):
for transformer_index in range(16):
self.set_model_patch_replace(patch, target, block_name, number, transformer_index)
return