From 8f315e1cd3b48eef0458519c124101a9062acd6a Mon Sep 17 00:00:00 2001 From: Thomas Ward Date: Fri, 31 May 2024 09:48:35 -0400 Subject: [PATCH] Revert "GenerationParameters: Condense SAMPLER,SCHEDULER and string equivs into single items with multiple types" This reverts commit f1d6a0a47a73e85293dbad09484b593ffcb95ce9. --- generation.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/generation.py b/generation.py index b1f1acd..502b655 100644 --- a/generation.py +++ b/generation.py @@ -118,9 +118,10 @@ class TWCUI_Util_GenerationParameters(BaseNode): } } - RETURN_TYPES = ("LATENT", "COMBO,STRING", "COMBO,STRING", "INT", - "INT", "INT", "FLOAT", "INT") - RETURN_NAMES = ("LATENT", "sampler_name", "scheduler", "width", "height", "steps", "cfg", "seed") + 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") def process(self, image_width: int, image_height: int, sampling_steps: int, cfg: float, sampler_name: str, scheduler_name: str, seed: int) -> tuple: