Add node for advanced merging by block.

This commit is contained in:
comfyanonymous
2023-06-20 22:11:01 -04:00
committed by GitHub
parent e881b1d93b
commit 8c6a8fe601
+30
View File
@@ -0,0 +1,30 @@
import comfy_extras.nodes_model_merging
class ModelMergeBlockNumber(comfy_extras.nodes_model_merging.ModelMergeBlocks):
@classmethod
def INPUT_TYPES(s):
arg_dict = { "model1": ("MODEL",),
"model2": ("MODEL",)}
argument = ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01})
arg_dict["time_embed."] = argument
arg_dict["label_emb."] = argument
for i in range(12):
arg_dict["input_blocks.{}".format(i)] = argument
for i in range(3):
arg_dict["middle_block.{}".format(i)] = argument
for i in range(12):
arg_dict["output_blocks.{}".format(i)] = argument
arg_dict["out."] = argument
return {"required": arg_dict}
NODE_CLASS_MAPPINGS = {
"ModelMergeBlockNumber": ModelMergeBlockNumber,
}