| 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 = {} |