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: