ModelMergeBlockNumber

Documentation

Class name: ModelMergeBlockNumber
Category: advanced/model_merging
Output node: False

This node specializes in merging two models by blending their components based on specified ratios for different parts of the models. It allows for fine-grained control over the merging process by enabling the adjustment of blend ratios for various blocks within the models.

Input types

Required

model1
- The first model to be merged. It serves as the base model onto which patches from the second model are applied.
- Comfy dtype: MODEL
- Python dtype: comfy.model_base.ModelBase
model2
- The second model to be merged. Key patches from this model are applied to the first model based on specified ratios.
- Comfy dtype: MODEL
- Python dtype: comfy.model_base.ModelBase
time_embed.
- Specifies the blend ratio for the time embedding components of the models.
- Comfy dtype: FLOAT
- Python dtype: float
label_emb.
- Specifies the blend ratio for the label embedding components of the models.
- Comfy dtype: FLOAT
- Python dtype: float
input_blocks.i.
- Specifies the blend ratio for each of the input blocks. The index 'i' ranges from 0 to 11, allowing for individual adjustment of 12 input blocks.
- Comfy dtype: FLOAT
- Python dtype: float
middle_block.i.
- Specifies the blend ratio for each of the middle blocks. The index 'i' ranges from 0 to 2, targeting 3 middle blocks for adjustment.
- Comfy dtype: FLOAT
- Python dtype: float
output_blocks.i.
- Specifies the blend ratio for each of the output blocks. The index 'i' ranges from 0 to 11, enabling individual adjustment of 12 output blocks.
- Comfy dtype: FLOAT
- Python dtype: float
out.
- Specifies the blend ratio for the final output components of the models.
- Comfy dtype: FLOAT
- Python dtype: float

Output types

model
- Comfy dtype: MODEL
- The result of merging two models based on the specified blend ratios for various components.
- Python dtype: comfy.model_base.ModelBase

Usage tips

Infra type: CPU
Common nodes: unknown

Source code

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}

3.0 KiB Raw Permalink Blame History