INNER CODE UNIT · Python

INPUT_TYPES

Extraltodeus/ComfyUI-AutomaticCFG · experimental_temperature.py:129

    def INPUT_TYPES(cls):
        return {"required": { "clip": ("CLIP",),
                              "Temperature": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 10.0, "step": 0.01}),
                              }}
    
    RETURN_TYPES = ("CLIP",)
    FUNCTION = "patch"
    CATEGORY = "model_patches/Automatic_CFG/Standalone_temperature_patches"
    
    def patch(self, clip, Temperature):
        def custom_optimized_attention(device, mask=None, small_input=True):
            return temperature_patcher(Temperature).attention_basic_with_temperature
        
        def new_forward(self, x, mask=None, intermediate_output=None):
            optimized_attention = custom_optimized_attention(x.device, mask=mask is not None, small_input=True)

            if intermediate_output is not None:
                if intermediate_output < 0:

View source record →