chunzhi/comfyui-deploy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: make external lora and checkpoint wildcard

Browse Source
This commit is contained in:
bennykok 2024-07-26 17:39:40 -07:00
parent 4f1a80fb64
commit e0eae1068b
2 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy-nodes/external_checkpoints.py
View File

@ -5,6 +5,12 @@ import torch
import folder_paths import folder_paths
from tqdm import tqdm from tqdm import tqdm
class AnyType(str):
def __ne__(self, __value: object) -> bool:
return False
WILDCARD = AnyType("*")
class ComfyUIDeployExternalCheckpoint: class ComfyUIDeployExternalCheckpoint:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -20,7 +26,7 @@ class ComfyUIDeployExternalCheckpoint:
} }
} }
RETURN_TYPES = (folder_paths.get_filename_list("checkpoints"),) RETURN_TYPES = (WILDCARD,)
RETURN_NAMES = ("path",) RETURN_NAMES = ("path",)
FUNCTION = "run" FUNCTION = "run"

7
comfy-nodes/external_lora.py
View File

@ -4,6 +4,11 @@ import numpy as np
import torch import torch
import folder_paths import folder_paths
class AnyType(str):
def __ne__(self, __value: object) -> bool:
return False
WILDCARD = AnyType("*")
class ComfyUIDeployExternalLora: class ComfyUIDeployExternalLora:
@classmethod @classmethod
@ -20,7 +25,7 @@ class ComfyUIDeployExternalLora:
}, },
} }
RETURN_TYPES = (folder_paths.get_filename_list("loras"),) RETURN_TYPES = (WILDCARD,)
RETURN_NAMES = ("path",) RETURN_NAMES = ("path",)
FUNCTION = "run" FUNCTION = "run"