jbetker commited on
Commit
5a95c34
1 Parent(s): 39ec1b0

fix warning

Browse files
Files changed (1) hide show
  1. tortoise/utils/stft.py +1 -1
tortoise/utils/stft.py CHANGED
@@ -116,7 +116,7 @@ class STFT(torch.nn.Module):
116
  assert(filter_length >= win_length)
117
  # get window and zero center pad it to filter_length
118
  fft_window = get_window(window, win_length, fftbins=True)
119
- fft_window = pad_center(fft_window, filter_length)
120
  fft_window = torch.from_numpy(fft_window).float()
121
 
122
  # window the bases
 
116
  assert(filter_length >= win_length)
117
  # get window and zero center pad it to filter_length
118
  fft_window = get_window(window, win_length, fftbins=True)
119
+ fft_window = pad_center(fft_window, size=filter_length)
120
  fft_window = torch.from_numpy(fft_window).float()
121
 
122
  # window the bases