File size: 930 Bytes
c336648 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
from . import smea_sampling
from .smea_sampling import sample_euler_dy, sample_euler_smea_dy, sample_euler_negative, sample_euler_dy_negative
if smea_sampling.BACKEND == "ComfyUI":
if not smea_sampling.INITIALIZED:
from comfy.k_diffusion import sampling as k_diffusion_sampling
from comfy.samplers import SAMPLER_NAMES
setattr(k_diffusion_sampling, "sample_euler_dy", sample_euler_dy)
setattr(k_diffusion_sampling, "sample_euler_smea_dy", sample_euler_smea_dy)
setattr(k_diffusion_sampling, "sample_euler_negative", sample_euler_negative)
setattr(k_diffusion_sampling, "sample_euler_dy_negative", sample_euler_dy_negative)
SAMPLER_NAMES.append("euler_dy")
SAMPLER_NAMES.append("euler_smea_dy")
SAMPLER_NAMES.append("euler_negative")
SAMPLER_NAMES.append("euler_dy_negative")
smea_sampling.INITIALIZED = True
NODE_CLASS_MAPPINGS = {}
|