This commit is contained in:
flak (home)
2023-09-14 09:50:33 +03:00
parent c7f9bfb8ac
commit 66671237ab
+1 -1
View File
@@ -49,5 +49,5 @@ class CircularVAEDecode:
for layer in vae.first_stage_model.modules()
if isinstance(layer, torch.nn.Conv2d)
]:
layer.padding_mode = "circular"
layer.padding_mode = "zeros"
return result