From 4836e1f79281ece5a073bb958d98657b7b810146 Mon Sep 17 00:00:00 2001 From: James Betker Date: Mon, 2 May 2022 16:36:02 -0600 Subject: [PATCH] fix warning --- tortoise/utils/stft.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tortoise/utils/stft.py b/tortoise/utils/stft.py index 8de6bfb..f54eb96 100644 --- a/tortoise/utils/stft.py +++ b/tortoise/utils/stft.py @@ -116,7 +116,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, filter_length) + fft_window = pad_center(fft_window, size=filter_length) fft_window = torch.from_numpy(fft_window).float() # window the bases