From af0825f021d7f16e21f5366a22d739e85fe0c746 Mon Sep 17 00:00:00 2001 From: traugdor Date: Fri, 1 Nov 2024 16:16:34 -0500 Subject: [PATCH] Update nodes.py revert change to VAEDecode node. --- nodes.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/nodes.py b/nodes.py index d56d26383..8fe7bf3e7 100644 --- a/nodes.py +++ b/nodes.py @@ -281,13 +281,10 @@ class VAEDecode: DESCRIPTION = "Decodes latent images back into pixel space images." def decode(self, vae, samples, tile_size=512): - try: - images = vae.decode(samples["samples"]) - if len(images.shape) == 5: #Combine batches - images = images.reshape(-1, images.shape[-3], images.shape[-2], images.shape[-1]) - return (images, ) - except (MemoryError, RuntimeError) as e: - return (vae.decode_tiled(samples["samples"], tile_x=tile_size // 8, tile_y=tile_size // 8, ), ) + images = vae.decode(samples["samples"]) + if len(images.shape) == 5: #Combine batches + images = images.reshape(-1, images.shape[-3], images.shape[-2], images.shape[-1]) + return (images, ) class VAEDecodeTiled: @classmethod