From 42d23e24225ed45e407593d0dfe57777053e28f0 Mon Sep 17 00:00:00 2001
From: braintacles <braintacles@gmail.com>
Date: Wed, 6 Sep 2023 07:27:31 +0900
Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20return=20types?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 braintacles_nodes.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/braintacles_nodes.py b/braintacles_nodes.py
index 485bc1d..d7301b1 100644
--- a/braintacles_nodes.py
+++ b/braintacles_nodes.py
@@ -197,7 +197,7 @@ class VAEDecodePipe:
     @classmethod
     def INPUT_TYPES(s):
         return {"required": {"samples": ("LATENT", ), "vae": ("VAE", )}}
-    RETURN_TYPES = ("IMAGE",)
+    RETURN_TYPES = ("IMAGE","VAE",)
     FUNCTION = "decode"
 
     CATEGORY = "braintacles/latent"
@@ -212,7 +212,7 @@ class VAEDecodeTiledPipe:
         return {"required": {"samples": ("LATENT", ), "vae": ("VAE", ),
                              "tile_size": ("INT", {"default": 512, "min": 320, "max": 4096, "step": 64})
                              }}
-    RETURN_TYPES = ("IMAGE",)
+    RETURN_TYPES = ("IMAGE","VAE",)
     FUNCTION = "decode"
 
     CATEGORY = "braintacles/latent"
@@ -225,7 +225,7 @@ class VAEEncodePipe:
     @classmethod
     def INPUT_TYPES(s):
         return {"required": {"pixels": ("IMAGE", ), "vae": ("VAE", )}}
-    RETURN_TYPES = ("LATENT",)
+    RETURN_TYPES = ("LATENT","VAE",)
     FUNCTION = "encode"
 
     CATEGORY = "braintacles/latent"
@@ -252,7 +252,7 @@ class VAEEncodeTiledPipe:
         return {"required": {"pixels": ("IMAGE", ), "vae": ("VAE", ),
                              "tile_size": ("INT", {"default": 512, "min": 320, "max": 4096, "step": 64})
                              }}
-    RETURN_TYPES = ("LATENT",)
+    RETURN_TYPES = ("LATENT","VAE",)
     FUNCTION = "encode"
 
     CATEGORY = "braintacles/latent"