diff options
Diffstat (limited to 'modules/sd_hijack.py')
| -rw-r--r-- | modules/sd_hijack.py | 5 |
1 files changed, 5 insertions, 0 deletions
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()
|