From f5a243a89830e369dbf5e905fe4aae5791b1851f Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 9 May 2025 13:48:30 +0300
Subject: [PATCH] Update utils.py

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

diff --git a/utils.py b/utils.py
index 39a4c15..de88ead 100644
--- a/utils.py
+++ b/utils.py
@@ -66,11 +66,11 @@ def fourier_filter(x, scale_low=1.0, scale_high=1.5, freq_cutoff=20):
     x_freq = fft.fftshift(x_freq, dim=(-2, -1))
 
     # 2) Create a mask to scale frequencies differently
-    C, B, H, W = x_freq.shape
+    B, C, T, H, W = x_freq.shape
     crow, ccol = H // 2, W // 2
 
     # Initialize mask with high-frequency scaling factor
-    mask = torch.ones((C, B, H, W), device=device) * scale_high
+    mask = torch.ones((B, C, T, H, W), device=device) * scale_high
 
     # Apply low-frequency scaling factor to center region
     mask[