22 lines
633 B
Python
22 lines
633 B
Python
from . import tcd_sampling
|
|
from .tcd_sampling import sample_tcd, sample_tcd_euler_a
|
|
from .tcd_scheduler import TCDScheduler
|
|
|
|
|
|
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 = {
|
|
"TCDScheduler": TCDScheduler
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS']
|