fffiloni commited on
Commit
3bebdb5
·
verified ·
1 Parent(s): 28d40df

Update audioldm/audio/stft.py

Browse files
Files changed (1) hide show
  1. 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