Spaces:

fffiloni
/

vta-ldm

Running on Zero

fffiloni commited on Jul 25, 2024

Commit

28d40df

verified ·

1 Parent(s): ca81f61

Update audioldm/audio/stft.py

Files changed (1) hide show

audioldm/audio/stft.py CHANGED Viewed

@@ -39,7 +39,7 @@ class STFT(torch.nn.Module):
             assert filter_length >= win_length
             # get window and zero center pad it to filter_length
             fft_window = get_window(window, win_length, fftbins=True)
-            fft_window = pad_center(fft_window)
             fft_window = torch.from_numpy(fft_window).float()
             # window the bases

             assert filter_length >= win_length
             # get window and zero center pad it to filter_length
             fft_window = get_window(window, win_length, fftbins=True)
+            fft_window = pad_center(fft_window, filter_length)
             fft_window = torch.from_numpy(fft_window).float()
             # window the bases