From 543f39ebf27490a9129c030e916bcb86726b5332 Mon Sep 17 00:00:00 2001
From: blepping <blepping@tuta.io>
Date: Mon, 3 Mar 2025 04:40:39 -0700
Subject: [PATCH] Fix dims order in pyramid noise generator

---
 py/noise_generation.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/py/noise_generation.py b/py/noise_generation.py
index 48aff90..3a86d25 100644
--- a/py/noise_generation.py
+++ b/py/noise_generation.py
@@ -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,