14 lines
411 B
Python
14 lines
411 B
Python
from . import restart_sampling
|
|
from .restart_sampling import sample_restart
|
|
|
|
|
|
if not restart_sampling.INITIALIZED:
|
|
from comfy.k_diffusion import sampling as k_diffusion_sampling
|
|
from comfy.samplers import SAMPLER_NAMES
|
|
|
|
setattr(k_diffusion_sampling, "sample_restart", sample_restart)
|
|
SAMPLER_NAMES.append("restart")
|
|
restart_sampling.INITIALIZED = True
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {} |