55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
class vaeSearch:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"VAE Search": ("STRING", {
|
|
"tooltip": "Searchg For A Specific VAE By Name Or Civit AIR Code (eg: ClearVAE).",
|
|
}),
|
|
"Model Architecture": ([
|
|
"All",
|
|
"FLUX.1-Schnell",
|
|
"FLUX.1-Dev",
|
|
"Pony",
|
|
"SD 1.5",
|
|
"SD 1.5 Hyper",
|
|
"SD 1.5 LCM",
|
|
"SD 3",
|
|
"SDXL 1.0",
|
|
"SDXL 1.0 LCM",
|
|
"SDXL Distilled",
|
|
"SDXL Hyper",
|
|
"SDXL Lightning",
|
|
"SDXL Turbo",
|
|
], {
|
|
"tooltip": "Choose VAE's Model Architecture To Filter Out The Results.",
|
|
"default": "SD 1.5",
|
|
}),
|
|
"VAEList": ([
|
|
"civitai:23906@28569 (kl-f8-anime2 VAE)",
|
|
"civitai:22354@88156 (ClearVAE(SD1.5))",
|
|
"civitai:276082@311162 (vae-ft-mse-840000-ema-pruned | 840000 | 840k SD1.5 VAE)",
|
|
"civitai:70248@83798 (Color101 VAE)",
|
|
"civitai:22354@26689 (ClearVAE (SD1.5))",
|
|
"civitai:88390@94036 (difConsistency RAW VAE (Pack))",
|
|
], {
|
|
"tooltip": "VAE Results Will Show UP Here So You Could Choose From.",
|
|
"default": "civitai:23906@28569 (kl-f8-anime2 VAE)",
|
|
})
|
|
},
|
|
}
|
|
|
|
DESCRIPTION = "Directly Search and Connect VAE's to Runware Inference Nodes In ComfyUI."
|
|
FUNCTION = "vaeSearch"
|
|
RETURN_TYPES = ("RUNWAREVAE",)
|
|
RETURN_NAMES = ("Runware VAE",)
|
|
CATEGORY = "Runware"
|
|
|
|
@classmethod
|
|
def VALIDATE_INPUTS(cls, VAEList):
|
|
return True
|
|
|
|
def vaeSearch(self, **kwargs):
|
|
currentModel = kwargs.get("VAEList")
|
|
vaeAIRCode = currentModel.split(" ")[0]
|
|
return (vaeAIRCode,) |