INNER CODE UNIT · Python

ExperimentalTemperaturePatch

Extraltodeus/ComfyUI-AutomaticCFG · experimental_temperature.py:88

class ExperimentalTemperaturePatch:
    @classmethod
    def INPUT_TYPES(s):
        required_inputs = {f"{key}_{layer}": ("BOOLEAN", {"default": False}) for key, layers in s.TOGGLES.items() for layer in layers}
        required_inputs["model"] = ("MODEL",)
        required_inputs["Temperature"]  = ("FLOAT", {"default": 1.0, "min": 0.0, "max": 10.0, "step": 0.01, "round": 0.01})
        required_inputs["Attention"]    = (["both","self","cross"],)
        return {"required": required_inputs}
    
    TOGGLES = {}
    RETURN_TYPES = ("MODEL","STRING",)
    RETURN_NAMES = ("Model","String",)
    FUNCTION = "patch"

    CATEGORY = "model_patches/Automatic_CFG/Standalone_temperature_patches"

    def patch(self, model, Temperature, Attention, **kwargs):
        m = model.clone()

View source record →