Fix dims order in pyramid noise generator

2025-03-03 04:40:39 -07:00
parent 8097f26863
commit 543f39ebf2
1 changed files with 2 additions and 2 deletions
@@ -626,14 +626,14 @@ class PyramidNoiseGenerator(FramesToChannelsNoiseGenerator):
                torch.randn(
                    b,
                    c,
-                    w,
                    h,
+                    w,
                    device=noise.device,
                    layout=noise.layout,
                    dtype=noise.dtype,
                ),
-                orig_h,
                orig_w,
+                orig_h,
                mode=self.upscale_mode,
            ).mul_(
                self.discount**i,