mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-11 03:58:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add missing samplers to KSamplerSelect.

Browse Source
This commit is contained in:
comfyanonymous
2023-09-28 00:17:03 -04:00
parent 1d7dfc07d5
commit d234ca558a
2 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_custom_sampler.py
View File

@@ -28,7 +28,7 @@ class KSamplerSelect:
@classmethod
def INPUT_TYPES(s):
return {"required":
{"sampler_name": (comfy.samplers.KSAMPLER_NAMES, ),
{"sampler_name": (comfy.samplers.SAMPLER_NAMES, ),
}
}
RETURN_TYPES = ("SAMPLER",)
@@ -37,7 +37,7 @@ class KSamplerSelect:
FUNCTION = "get_sampler"
def get_sampler(self, sampler_name):
sampler = comfy.samplers.ksampler(sampler_name)()
sampler = comfy.samplers.sampler_class(sampler_name)()
return (sampler, )
class SamplerCustom: