ModelMergeAdd

Documentation

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

The ModelMergeAdd node is designed for merging two models by adding key patches from one model to another. This process involves cloning the first model and then applying patches from the second model, allowing for the combination of features or behaviors from both models.

Input types

Required

model1
- The first model to be cloned and to which patches from the second model will be added. It serves as the base model for the merging process.
- Comfy dtype: MODEL
- Python dtype: comfy.model_base.Model
model2
- The second model from which key patches are extracted and added to the first model. It contributes additional features or behaviors to the merged model.
- Comfy dtype: MODEL
- Python dtype: comfy.model_base.Model

Output types

model
- Comfy dtype: MODEL
- The result of merging two models by adding key patches from the second model to the first. This merged model combines features or behaviors from both models.
- Python dtype: comfy.model_base.Model

Usage tips

Infra type: GPU
Common nodes: unknown

Source code

class ModelAdd:
    @classmethod
    def INPUT_TYPES(s):
        return {"required": { "model1": ("MODEL",),
                              "model2": ("MODEL",),
                              }}
    RETURN_TYPES = ("MODEL",)
    FUNCTION = "merge"

    CATEGORY = "advanced/model_merging"

    def merge(self, model1, model2):
        m = model1.clone()
        kp = model2.get_key_patches("diffusion_model.")
        for k in kp:
            m.add_patches({k: kp[k]}, 1.0, 1.0)
        return (m, )

1.7 KiB Raw Permalink Blame History