INNER CODE UNIT · Python

cfg_result

Extraltodeus/ComfyUI-AutomaticCFG · nodes.py:67

        cfg_result = uncond_pred + (cond_pred - uncond_pred) * cond_scale

    for fn in model_options.get("sampler_post_cfg_function", []):
        args = {"denoised": cfg_result, "cond": cond_copy, "uncond": uncond_copy, "model": model, "uncond_denoised": uncond_pred, "cond_denoised": cond_pred,
                "sigma": timestep, "model_options": model_options, "input": x}
        cfg_result = fn(args)
        
    return cfg_result

def monkey_patching_comfy_sampling_function():
    global original_sampling_function

    if original_sampling_function is None:
        original_sampling_function = comfy.samplers.sampling_function
    # Make sure to only patch once
    if hasattr(comfy.samplers.sampling_function, '_automatic_cfg_decorated'):
        return
    comfy.samplers.sampling_function = sampling_function_patched

View source record →