add Sampler_names

__init__.py

@@ -557,7 +557,7 @@ def new_add_routes(self):
 from .nodes.Clipseg import CLIPSeg,CombineMasks
 from .nodes.ChatGPT import ChatGPTNode,ShowTextForGPT,CharacterInText
 from .nodes.Audio import GamePal,SpeechRecognition,SpeechSynthesis
-from .nodes.Utils import CreateCkptNames,CreateSeedNode,TESTNODE_,AppInfo,IntNumber,FloatSlider,TextInput,ColorInput,FontInput,TextToNumber,DynamicDelayProcessor,LimitNumber,SwitchByIndex,MultiplicationNode
+from .nodes.Utils import CreateSampler_names,CreateCkptNames,CreateSeedNode,TESTNODE_,AppInfo,IntNumber,FloatSlider,TextInput,ColorInput,FontInput,TextToNumber,DynamicDelayProcessor,LimitNumber,SwitchByIndex,MultiplicationNode
 from .nodes.Mask import OutlineMask,FeatheredMask
 
 
@@ -619,7 +619,8 @@ def new_add_routes(self):
     "OutlineMask":OutlineMask,
     "JoinWithDelimiter":JoinWithDelimiter,
     "Seed_":CreateSeedNode,
-    "CkptNames_":CreateCkptNames
+    "CkptNames_":CreateCkptNames,
+    "SamplerNames_":CreateSampler_names
     # "LaMaInpainting":LaMaInpainting
     # "GamePal":GamePal
 }

data/extension-node-map.json

@@ -4780,6 +4780,7 @@
             "Color",
             "Seed_",
             "CkptNames_",
+            "SamplerNames_",
             "CombineMasks_",
             "EnhanceImage",
             "GradientImage",

nodes/Utils.py

@@ -7,6 +7,8 @@
 import matplotlib.font_manager as fm
 import torch
 
+
+
 def recursive_search(directory, excluded_dir_names=None):
     if not os.path.isdir(directory):
         return [], {}
@@ -765,4 +767,33 @@ def INPUT_TYPES(cls):
     def run(self, ckpt_names):
         ckpt_names=ckpt_names.split('\n')
         ckpt_names = [name for name in ckpt_names if name.strip()]
-        return (ckpt_names,)
+        return (ckpt_names,)
+
+
+class CreateSampler_names:
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "sampler_names": ("STRING",{"multiline": True,"default": "\n".join(comfy.samplers.KSampler.SAMPLERS),"dynamicPrompts": False}),
+            }
+        }
+
+    RETURN_TYPES = (any_type,)
+    RETURN_NAMES = ("sampler_names",)
+
+    INPUT_IS_LIST = False
+    OUTPUT_IS_LIST = (True,)
+
+    # OUTPUT_NODE = True
+    FUNCTION = "run"
+
+    CATEGORY = "♾️Mixlab/Utils"
+
+    def run(self, sampler_names):
+        sampler_names=sampler_names.split('\n')
+        sampler_names = [name for name in sampler_names if name.strip()]
+        return (sampler_names,)

web/javascript/smart_connect.js

@@ -22,6 +22,12 @@ const smart_connect_config_input = [
     node_widget_name: 'ckpt_name',
     inputNodeName: 'CkptNames_',
     inputNode_output_name: 'ckpt_names'
+  },
+  {
+    node_type: 'KSampler',
+    node_widget_name: 'sampler_name',
+    inputNodeName: 'SamplerNames_',
+    inputNode_output_name: 'sampler_names'
   }
 ]