From f1d6a0a47a73e85293dbad09484b593ffcb95ce9 Mon Sep 17 00:00:00 2001
From: Thomas Ward <teward@thomas-ward.net>
Date: Thu, 30 May 2024 19:21:03 -0400
Subject: [PATCH] GenerationParameters: Condense SAMPLER,SCHEDULER and string
 equivs into single items with multiple types

---
 generation.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/generation.py b/generation.py
index 502b655..b1f1acd 100644
--- a/generation.py
+++ b/generation.py
@@ -118,10 +118,9 @@ class TWCUI_Util_GenerationParameters(BaseNode):
             }
         }
 
-    RETURN_TYPES = ("LATENT", comfy.samplers.KSampler.SAMPLERS, comfy.samplers.KSampler.SCHEDULERS, "INT",
-                    "INT", "INT", "FLOAT", "STRING", "STRING", "INT")
-    RETURN_NAMES = ("LATENT", "SAMPLER", "SCHEDULER", "width", "height", "steps", "cfg", "sampler_name",
-                    "scheduler", "seed")
+    RETURN_TYPES = ("LATENT", "COMBO,STRING", "COMBO,STRING", "INT",
+                    "INT", "INT", "FLOAT", "INT")
+    RETURN_NAMES = ("LATENT", "sampler_name", "scheduler", "width", "height", "steps", "cfg", "seed")
 
     def process(self, image_width: int, image_height: int, sampling_steps: int,
                 cfg: float, sampler_name: str, scheduler_name: str, seed: int) -> tuple: