diff --git a/ldm_patched/pfn/architecture/RRDB.py b/ldm_patched/pfn/architecture/RRDB.py index 8d318b90..3e2b7222 100644 --- a/ldm_patched/pfn/architecture/RRDB.py +++ b/ldm_patched/pfn/architecture/RRDB.py @@ -281,6 +281,7 @@ class RRDBNet(nn.Module): return max(*nbs) + 1 def forward(self, x): + x = x.contiguous() if self.shuffle_factor: _, _, h, w = x.size() mod_pad_h = (