INNER CODE UNIT · Python

CLIPTemperaturePatchDual

Extraltodeus/ComfyUI-AutomaticCFG · experimental_temperature.py:167

class CLIPTemperaturePatchDual:
    @classmethod
    def INPUT_TYPES(cls):
        return {"required": { "clip": ("CLIP",),
                              "Temperature": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 10.0, "step": 0.01}),
                              "CLIP_Model": (["clip_g","clip_l","both"],),
                              }}
    
    RETURN_TYPES = ("CLIP",)
    FUNCTION = "patch"
    CATEGORY = "model_patches/Automatic_CFG/Standalone_temperature_patches"
    
    def patch(self, clip, Temperature, CLIP_Model):
        def custom_optimized_attention(device, mask=None, small_input=True):
            return temperature_patcher(Temperature, "CLIP").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)

View source record →