From 8c65bc0090a371e79b2b42542a687d71c4d9b227 Mon Sep 17 00:00:00 2001 From: ssit Date: Wed, 26 Jul 2023 18:33:37 -0400 Subject: [PATCH] Update supported samplers --- nodes.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index fe1f2ee..1f75b69 100644 --- a/nodes.py +++ b/nodes.py @@ -4,10 +4,14 @@ from .restart_sampling import restart_sampling, SCHEDULER_MAPPING def get_supported_samplers(): samplers = comfy.samplers.KSampler.SAMPLERS.copy() - samplers.remove("dpm_fast") - samplers.remove("dpm_adaptive") samplers.remove("uni_pc") samplers.remove("uni_pc_bh2") + + # SDE samplers cannot be used with restarts + samplers.remove("dpmpp_sde") + samplers.remove("dpmpp_sde_gpu") + samplers.remove("dpmpp_2m_sde") + samplers.remove("dpmpp_2m_sde_gpu") return samplers @@ -51,7 +55,7 @@ class KRestartSampler: "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}), "steps": ("INT", {"default": 20, "min": 1, "max": 10000}), "cfg": ("FLOAT", {"default": 8.0, "min": 0.0, "max": 100.0}), - "sampler_name": (comfy.samplers.KSampler.SAMPLERS, ), + "sampler_name": (get_supported_samplers(), ), "scheduler": (comfy.samplers.KSampler.SCHEDULERS, ), "positive": ("CONDITIONING", ), "negative": ("CONDITIONING", ),