extensions/ComfyUI-TCD-Sampler/__init__.py

from . import tcd_sampling
from .tcd_sampling import sample_tcd, sample_tcd_euler_a


if not tcd_sampling.INITIALIZED:
    from comfy.k_diffusion import sampling as k_diffusion_sampling
    from comfy.samplers import SAMPLER_NAMES

    setattr(k_diffusion_sampling, "sample_tcd", sample_tcd)
    setattr(k_diffusion_sampling, "sample_tcd_euler_a", sample_tcd_euler_a)
    SAMPLER_NAMES.append("tcd")
    SAMPLER_NAMES.append("tcd_euler_a")
    tcd_sampling.INITIALIZED = True


NODE_CLASS_MAPPINGS = {}