From 59841afbd8ce6c194eb354e5a78ced103c77bda7 Mon Sep 17 00:00:00 2001
From: Lee <flyingfireco@gmail.com>
Date: Sun, 13 Aug 2023 17:09:15 -0400
Subject: [PATCH] added circular VAE Decoder

---
 tilingSampler.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/tilingSampler.py b/tilingSampler.py
index 7de2475..cb4294e 100644
--- a/tilingSampler.py
+++ b/tilingSampler.py
@@ -100,7 +100,22 @@ class Tiled_KSampler:
         return nodes.common_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent_image, denoise=denoise)
 
 
+class CircularVAEDecode:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": { "samples": ("LATENT", ), "vae": ("VAE", )}}
+    RETURN_TYPES = ("IMAGE",)
+    FUNCTION = "decode"
+
+    CATEGORY = "latent"
+
+    def decode(self, vae, samples):
+        for layer in [layer for layer in vae.first_stage_model.modules() if isinstance(layer, torch.nn.Conv2d)]:
+            layer.padding_mode = 'circular'
+        return (vae.decode(samples["samples"]), )
+
 NODE_CLASS_MAPPINGS = {
         "Tiled KSampler": Tiled_KSampler,
         "Asymmetric Tiled KSampler": Asymmetric_Tiled_KSampler,
+        "Circular VAEDecode": CircularVAEDecode,
 }
\ No newline at end of file