nodes/ssitu-ComfyUI_restart_sampling
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update supported samplers

Browse Source
This commit is contained in:
ssit
2023-07-26 18:33:37 -04:00
parent 1c3ed8e7bb
commit 8c65bc0090
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nodes.py
+7 -3
View File
@@ -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", ),