diff --git a/ldm_patched/pfn/architecture/RRDB.py b/ldm_patched/pfn/architecture/RRDB.py index 8d318b90b8..3e2b722289 100644 --- a/ldm_patched/pfn/architecture/RRDB.py +++ b/ldm_patched/pfn/architecture/RRDB.py @@ -281,6 +281,7 @@ def get_num_blocks(self) -> int: return max(*nbs) + 1 def forward(self, x): + x = x.contiguous() if self.shuffle_factor: _, _, h, w = x.size() mod_pad_h = (