Spaces:
Sleeping
Sleeping
Update audioldm/audio/stft.py
Browse files- audioldm/audio/stft.py +1 -1
audioldm/audio/stft.py
CHANGED
@@ -39,7 +39,7 @@ class STFT(torch.nn.Module):
|
|
39 |
assert filter_length >= win_length
|
40 |
# get window and zero center pad it to filter_length
|
41 |
fft_window = get_window(window, win_length, fftbins=True)
|
42 |
-
fft_window = pad_center(fft_window, filter_length)
|
43 |
fft_window = torch.from_numpy(fft_window).float()
|
44 |
|
45 |
# window the bases
|
|
|
39 |
assert filter_length >= win_length
|
40 |
# get window and zero center pad it to filter_length
|
41 |
fft_window = get_window(window, win_length, fftbins=True)
|
42 |
+
fft_window = pad_center(fft_window, size=filter_length)
|
43 |
fft_window = torch.from_numpy(fft_window).float()
|
44 |
|
45 |
# window the bases
|