From 50c82c06f2147cec45960bed20e0f75de23d717d Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 30 Jan 2024 19:21:31 -0800 Subject: [PATCH] Update unet_patcher.py --- modules_forge/unet_patcher.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modules_forge/unet_patcher.py b/modules_forge/unet_patcher.py index dcc0c382..4f429bae 100644 --- a/modules_forge/unet_patcher.py +++ b/modules_forge/unet_patcher.py @@ -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