From f91d0c3d19ac2b849cd15a519f711f9281e782cd Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 5 Sep 2022 02:16:36 +0300
Subject: add an option to enable tiling image generation

---
 modules/sd_hijack.py | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'modules/sd_hijack.py')

diff --git a/modules/sd_hijack.py b/modules/sd_hijack.py
index 1dbdc9ce..9779c30c 100644
--- a/modules/sd_hijack.py
+++ b/modules/sd_hijack.py
@@ -243,8 +243,13 @@ class EmbeddingsWithFixes(torch.nn.Module):
         return inputs_embeds
 
 
+def add_circular_option_to_conv_2d():
+    conv2d_constructor = torch.nn.Conv2d.__init__
 
+    def conv2d_constructor_circular(self, *args, **kwargs):
+        return conv2d_constructor(self, *args, padding_mode='circular', **kwargs)
 
+    torch.nn.Conv2d.__init__ = conv2d_constructor_circular
 
 
 model_hijack = StableDiffusionModelHijack()
-- 
cgit v1.2.1