Added VAE, Embedding + Combiner nodes, Added LyCoris in the options for lora, Fixed AIR Code from CivitAI, Automatically add trigger words for lora and embeddings, Added More workflows and fixed some typos in readme file

2024-12-18 01:04:13 +02:00
parent 6bf9d37204
commit 5472ad876e
20 changed files with 362 additions and 26 deletions
@@ -0,0 +1,55 @@
+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,)