diff --git a/comfy-nodes/external_checkpoints.py b/comfy-nodes/external_checkpoints.py index ea9b6ec..fabc9a1 100644 --- a/comfy-nodes/external_checkpoints.py +++ b/comfy-nodes/external_checkpoints.py @@ -5,6 +5,12 @@ import torch import folder_paths from tqdm import tqdm +class AnyType(str): + def __ne__(self, __value: object) -> bool: + return False + +WILDCARD = AnyType("*") + class ComfyUIDeployExternalCheckpoint: @classmethod def INPUT_TYPES(s): @@ -20,7 +26,7 @@ class ComfyUIDeployExternalCheckpoint: } } - RETURN_TYPES = (folder_paths.get_filename_list("checkpoints"),) + RETURN_TYPES = (WILDCARD,) RETURN_NAMES = ("path",) FUNCTION = "run" diff --git a/comfy-nodes/external_lora.py b/comfy-nodes/external_lora.py index ad47d62..d981829 100644 --- a/comfy-nodes/external_lora.py +++ b/comfy-nodes/external_lora.py @@ -4,6 +4,11 @@ import numpy as np import torch import folder_paths +class AnyType(str): + def __ne__(self, __value: object) -> bool: + return False + +WILDCARD = AnyType("*") class ComfyUIDeployExternalLora: @classmethod @@ -20,7 +25,7 @@ class ComfyUIDeployExternalLora: }, } - RETURN_TYPES = (folder_paths.get_filename_list("loras"),) + RETURN_TYPES = (WILDCARD,) RETURN_NAMES = ("path",) FUNCTION = "run"