From 52b8b02bee392e437a4d897e9f4a249d604c7d41 Mon Sep 17 00:00:00 2001
From: Thomas Ward <teward@thomas-ward.net>
Date: Sat, 11 May 2024 19:01:59 -0400
Subject: [PATCH] Fix VAE Selector input types

---
 models.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/models.py b/models.py
index 3ba732a..86eee78 100644
--- a/models.py
+++ b/models.py
@@ -86,6 +86,14 @@ class HelperNodes_VAESelector(BaseNode):
             sd["vae_scale"] = torch.tensor(0.13025)
         return sd
 
+    @classmethod
+    def INPUT_TYPES(cls) -> dict:
+        return {
+            "required": {
+                "vae_name": (cls.vae_list(),)
+            }
+        }
+
     CATEGORY = f"{MODULE_CATEGORY}"
 
     RETURN_TYPES = ("VAE",)