From 61beecb5eeb35d86ec0db7e61fe9d0e83eb06552 Mon Sep 17 00:00:00 2001
From: XchanBik <xchanbik@gmail.com>
Date: Sat, 10 May 2025 13:44:12 +0200
Subject: [PATCH] Test input

---
 nodes.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index 6f41bbe..7551f68 100644
--- a/nodes.py
+++ b/nodes.py
@@ -5,11 +5,14 @@ class BridgeStoreNode:
     def INPUT_TYPES(cls):
         return {
             "required": {
+                "model": ("MODEL", ),            
                 "data": (["LATENT", "CONDITIONING", "IMAGE", "STRING", "ANY"],),
                 "bridge_id": ("STRING", {"default": "my_key"})
+            },
+            "optional": {
+                "vae": ("VAE",),
             }
         }
-
     RETURN_TYPES = ()
     FUNCTION = "store"