This commit is contained in:
flak (home)
2023-09-14 09:30:49 +03:00
parent 1b3b2819ec
commit b6530d9539
+1 -1
View File
@@ -17,7 +17,7 @@ class SeamlessTile:
FUNCTION = "run"
def run(self, model, tiling):
for m in model.model.modules:
for m in model.model.modules():
if isinstance(m, torch.nn.Conv2d):
if tiling == "enable":
m.padding_mode = "circular"