From 3cc3eb53d93cc1c6c763d3e536552b6169ae0095 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Tue, 20 Jun 2023 22:25:02 -0400
Subject: [PATCH] Fix possible issue.

---
 advanced_model_merging.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/advanced_model_merging.py b/advanced_model_merging.py
index 2a69acf..f1d0686 100644
--- a/advanced_model_merging.py
+++ b/advanced_model_merging.py
@@ -12,13 +12,13 @@ class ModelMergeBlockNumber(comfy_extras.nodes_model_merging.ModelMergeBlocks):
         arg_dict["label_emb."] = argument
 
         for i in range(12):
-            arg_dict["input_blocks.{}".format(i)] = argument
+            arg_dict["input_blocks.{}.".format(i)] = argument
 
         for i in range(3):
-            arg_dict["middle_block.{}".format(i)] = argument
+            arg_dict["middle_block.{}.".format(i)] = argument
 
         for i in range(12):
-            arg_dict["output_blocks.{}".format(i)] = argument
+            arg_dict["output_blocks.{}.".format(i)] = argument
 
         arg_dict["out."] = argument