forked from mrq/tortoise-tts
fix warning
This commit is contained in:
parent
39ec1b0db5
commit
5a95c34a01
|
@ -116,7 +116,7 @@ class STFT(torch.nn.Module):
|
||||||
assert(filter_length >= win_length)
|
assert(filter_length >= win_length)
|
||||||
# get window and zero center pad it to filter_length
|
# get window and zero center pad it to filter_length
|
||||||
fft_window = get_window(window, win_length, fftbins=True)
|
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()
|
fft_window = torch.from_numpy(fft_window).float()
|
||||||
|
|
||||||
# window the bases
|
# window the bases
|
||||||
|
|
Loading…
Reference in New Issue
Block a user