Class APGGuiderNode

stg.py:888–944

@comfy_node(name="APGGuider")
class APGGuiderNode:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "model": ("MODEL",),
                "positive": ("CONDITIONING",),
                "negative": ("CONDITIONING",),
                "cfg_scale": (
                    "FLOAT",
                    {"default": 1.0, "min": 0.0, "max": 100.0, "step": 0.01},
                ),
                "eta": (
                    "FLOAT",
                    {"default": 1.0, "min": 0.0, "max": 100.0, "step": 0.01},
                ),
                "norm_threshold": (
                    "FLOAT",
                    {"default": 0.0, "min": 0.0, "max": 100.0, "step": 0.01},
                ),
                "momentum_coefficient": (
                    "FLOAT",
                    {"default": -0.9, "min": -3.0, "max": 3.0, "step": 0.01},
                ),
            },
        }

    RETURN_TYPES = ("GUIDER",)

    FUNCTION = "get_guider"
    CATEGORY = "lightricks/LTXV"

    DESCRIPTION = """
    The APG Guider implements Adaptive Projected Guidance (APG).
    Reference: https://arxiv.org/abs/2410.02416.
    """

    def get_guider(
        self,
        model,
        positive,
        negative,
        cfg_scale,
        eta,
        norm_threshold,
        momentum_coefficient,
    ):
        guider = APGGuider(
            model,
            cfg_scale,
            eta,
            norm_threshold,
            momentum_coefficient,
        )
        guider.set_conds(positive, negative)
        guider.raw_conds = (positive, negative)
        return (guider,)