Fix possible issue.

2023-06-20 22:25:02 -04:00
parent 8c6a8fe601
commit 3cc3eb53d9
1 changed files with 3 additions and 3 deletions
@@ -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