From d9adb71a724c4814ff69498804fb741926bbbcec Mon Sep 17 00:00:00 2001 From: HERIUN Date: Tue, 19 Nov 2024 19:00:51 +0900 Subject: [PATCH] torch.stft return_complex=True warning --- openvoice/mel_processing.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/openvoice/mel_processing.py b/openvoice/mel_processing.py index 822d7f19..82ea6d21 100644 --- a/openvoice/mel_processing.py +++ b/openvoice/mel_processing.py @@ -68,10 +68,10 @@ def spectrogram_torch(y, n_fft, sampling_rate, hop_size, win_size, center=False) pad_mode="reflect", normalized=False, onesided=True, - return_complex=False, + return_complex=True, ) - spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6) + spec = torch.sqrt(spec.real.pow(2) + spec.imag.pow(2) + 1e-6) return spec @@ -172,12 +172,12 @@ def mel_spectrogram_torch( pad_mode="reflect", normalized=False, onesided=True, - return_complex=False, + return_complex=True, ) - spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6) + spec = torch.sqrt(spec.real.pow(2) + spec.imag.pow(2) + 1e-6) spec = torch.matmul(mel_basis[fmax_dtype_device], spec) spec = spectral_normalize_torch(spec) - return spec \ No newline at end of file + return spec