File size: 1,342 Bytes
899ba56 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | from . import smea
from .smea import sample_spawner_smea, sample_spawner_smea_beta, sample_spawner_smea_dyn_beta, sample_spawner_smea_dyn_beta1, sample_spawner_rk2_smea_d_clamp, sample_spawner_smea_dyn_exp
if smea.BACKEND == "ComfyUI":
if not smea.INITIALIZED:
from comfy.k_diffusion import sampling as k_diffusion_sampling
from comfy.samplers import SAMPLER_NAMES
setattr(k_diffusion_sampling, "sample_spawner_smea", sample_spawner_smea)
setattr(k_diffusion_sampling, "sample_spawner_smea_beta", sample_spawner_smea_beta)
setattr(k_diffusion_sampling, "sample_spawner_smea_dyn_beta", sample_spawner_smea_dyn_beta)
setattr(k_diffusion_sampling, "sample_spawner_smea_dyn_beta1", sample_spawner_smea_dyn_beta1)
setattr(k_diffusion_sampling, "sample_spawner_smea_dyn_exp", sample_spawner_smea_dyn_exp)
setattr(k_diffusion_sampling, "sample_spawner_rk2_smea_d_clamp", sample_spawner_rk2_smea_d_clamp)
SAMPLER_NAMES.append("spawner_smea")
SAMPLER_NAMES.append("spawner_smea_beta")
SAMPLER_NAMES.append("spawner_smea_dyn_beta")
SAMPLER_NAMES.append("spawner_smea_dyn_beta1")
SAMPLER_NAMES.append("spawner_smea_dyn_exp")
SAMPLER_NAMES.append("spawner_rk2_smea_d_clamp")
smea.INITIALIZED = True
NODE_CLASS_MAPPINGS = {}
|