From e8ca90fe35c1e12a3bfa3e277f732ff62e6b9c61 Mon Sep 17 00:00:00 2001 From: Rhys Yang Date: Fri, 7 Jun 2024 01:32:18 +0800 Subject: [PATCH] Change the `Type Converter` node to a node that converts any type to a * type --- nodes.py | 31 ++++++------------------------- 1 file changed, 6 insertions(+), 25 deletions(-) diff --git a/nodes.py b/nodes.py index 6dd5b13..cdff623 100644 --- a/nodes.py +++ b/nodes.py @@ -1003,41 +1003,22 @@ class SDTypeConverter: return { "required": {}, "optional": { - "model_name": ( - folder_paths.get_filename_list("checkpoints"), + "any_type_input": ( + any_type, {"forceInput": True}, ), - "sampler_name": ( - comfy.samplers.KSampler.SAMPLERS, - {"forceInput": True}, - ), - "scheduler": (comfy.samplers.KSampler.SCHEDULERS, {"forceInput": True}), }, } - RETURN_TYPES = ( - "STRING", - "STRING", - "STRING", - ) + RETURN_TYPES = (any_type,) - RETURN_NAMES = ( - "MODEL_NAME_STR", - "SAMPLER_NAME_STR", - "SCHEDULER_STR", - ) + RETURN_NAMES = ("ANY_TYPE_OUTPUT",) FUNCTION = "convert_string" CATEGORY = "SD Prompt Reader" - def convert_string( - self, model_name: str = "", sampler_name: str = "", scheduler: str = "" - ): - return ( - model_name, - sampler_name, - scheduler, - ) + def convert_string(self, any_type_input: str = ""): + return (any_type_input,) class SDBatchLoader: