VAE circular padding

This commit is contained in:
AI-Casanova
2024-02-24 21:56:48 -06:00
committed by Vladimir Mandic
parent a13335a657
commit 316b74cd15
+2
View File
@@ -41,6 +41,8 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
try:
for layer in [layer for layer in shared.sd_model.unet.modules() if type(layer) is torch.nn.Conv2d]:
layer.padding_mode = 'circular' if enable else 'zeros'
for layer in [layer for layer in shared.sd_model.vae.modules() if type(layer) is torch.nn.Conv2d]:
layer.padding_mode = 'circular' if enable else 'zeros'
except Exception as e:
debug(f"Diffusers tiling failed: {e}")