INNER CODE UNIT · Python
monkey_patching_comfy_sampling_function
Extraltodeus/ComfyUI-AutomaticCFG · nodes.py:76
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
comfy.samplers.sampling_function._automatic_cfg_decorated = True # flag to check monkey patch
def make_sampler_pre_cfg_automatic_cfg_function(minimum_sigma_to_disable_uncond=0, maximum_sigma_to_enable_uncond=1000000, disabled_cond_start=10000,disabled_cond_end=10000):
def sampler_pre_cfg_automatic_cfg_function(sigma, uncond, cond, cond_scale, **kwargs):
if sigma[0] < minimum_sigma_to_disable_uncond or sigma[0] > maximum_sigma_to_enable_uncond:
uncond = None
if sigma[0] <= disabled_cond_start and sigma[0] > disabled_cond_end:
cond = None
return uncond, cond, cond_scale