From 0f0cae2f3b88a9146f32d47b13a3c7492ce1ad2f Mon Sep 17 00:00:00 2001
From: Won-Kyu Park <wkpark@gmail.com>
Date: Sat, 21 Oct 2023 22:56:06 +0900
Subject: [PATCH] HOT fix: fix middle_block string to get all possible
 middle_block keys

---
 scripts/model_mixer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/model_mixer.py b/scripts/model_mixer.py
index e6f563a..e42aca5 100644
--- a/scripts/model_mixer.py
+++ b/scripts/model_mixer.py
@@ -324,7 +324,7 @@ def _all_blocks(isxl=False):
     blocks = [ base_prefix ]
     for i in range(0, BLOCKLEN):
         blocks.append(f"input_blocks.{i}.")
-    blocks.append("middle_block.1.")
+    blocks.append("middle_block.")
     for i in range(0, BLOCKLEN):
         blocks.append(f"output_blocks.{i}.")
     return blocks