Adjust Weight [All◆Add] 🎭🅐🅓

Documentation

Class name: ADE_AdjustWeightAllAdd
Category: Animate Diff 🎭🅐🅓/ad settings/weight adjust
Output node: False

The ADE_AdjustWeightAllAdd node is designed to adjust the weights of various components in an animation diffusion model by adding specified values. This adjustment process is crucial for fine-tuning the animation generation process, allowing for more precise control over the animation's appearance and behavior.

Input types

Required

all_ADD
- Specifies the value to be added to all weight components, enabling a uniform adjustment across the model. This parameter allows for a broad modification of the model's behavior.
- Comfy dtype: FLOAT
- Python dtype: float
print_adjustment
- A boolean flag that, when set to true, enables the printing of the adjustment details for debugging or informational purposes.
- Comfy dtype: BOOLEAN
- Python dtype: bool

Optional

prev_weight_adjust
- An optional parameter that allows for the chaining of weight adjustments by providing a previous AdjustGroup object. This facilitates cumulative adjustments over multiple steps.
- Comfy dtype: WEIGHT_ADJUST
- Python dtype: AdjustGroup

Output types

weight_adjust
- Comfy dtype: WEIGHT_ADJUST
- Returns an updated AdjustGroup object that encapsulates the new weight adjustments, reflecting the changes made to the model's weights.
- Python dtype: AdjustGroup

Usage tips

Infra type: CPU
Common nodes: unknown

Source code

class WeightAdjustAllAddNode:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "all_ADD": ("FLOAT", {"default": 0.0, "min": -2.0, "max": 2.0, "step": 0.000001}),
                "print_adjustment": ("BOOLEAN", {"default": False}),
            },
            "optional": {
                "prev_weight_adjust": ("WEIGHT_ADJUST",),
            }
        }
    
    RETURN_TYPES = ("WEIGHT_ADJUST",)
    CATEGORY = "Animate Diff 🎭🅐🅓/ad settings/weight adjust"
    FUNCTION = "get_weight_adjust"

    def get_weight_adjust(self, all_ADD: float, print_adjustment: bool, prev_weight_adjust: AdjustGroup=None):
        if prev_weight_adjust is None:
            prev_weight_adjust = AdjustGroup()
        prev_weight_adjust = prev_weight_adjust.clone()
        adjust = AdjustWeight(
            all_ADD=all_ADD,
            print_adjustment=print_adjustment
        )
        prev_weight_adjust.add(adjust)
        return (prev_weight_adjust,)

2.6 KiB Raw Permalink Blame History