diff --git a/modules_forge/unet_patcher.py b/modules_forge/unet_patcher.py index ed2bfb26..8e2b6f82 100644 --- a/modules_forge/unet_patcher.py +++ b/modules_forge/unet_patcher.py @@ -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