nodes/kijai-ComfyUI-HunyuanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update utils.py

Browse Source
This commit is contained in:
kijai
2025-05-09 13:48:30 +03:00
parent 00c8864900
commit f5a243a898
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils.py
+2 -2
View File
@@ -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)) x_freq = fft.fftshift(x_freq, dim=(-2, -1))
# 2) Create a mask to scale frequencies differently # 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 crow, ccol = H // 2, W // 2
# Initialize mask with high-frequency scaling factor # 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 # Apply low-frequency scaling factor to center region
mask[ mask[