From f9c1e11851bb064603548ccb3690f4309f005abe Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 25 Oct 2024 21:49:48 +0300
Subject: [PATCH] diffusers backwards compatibility on tiled encode

for some reason it's tuple in 0.31.0
---
 mz_enable_vae_encode_tiling.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mz_enable_vae_encode_tiling.py b/mz_enable_vae_encode_tiling.py
index a038bec..544a649 100644
--- a/mz_enable_vae_encode_tiling.py
+++ b/mz_enable_vae_encode_tiling.py
@@ -81,6 +81,8 @@ def tiled_encode(self, x: torch.Tensor) -> torch.Tensor:
                 ]
                 
                 tile = self.encoder(tile)
+                if not isinstance(tile, tuple):
+                    tile = (tile,)
                 if self.quant_conv is not None:
                     tile = self.quant_conv(tile)
                 time.append(tile[0])