HOT fix: fix middle_block string to get all possible middle_block keys

wkpark · Oct 21, 2023 · 0f0cae2 · 0f0cae2
1 parent bfb79fa
commit 0f0cae2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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