diff --git a/GPT_SoVITS/module/mel_processing.py b/GPT_SoVITS/module/mel_processing.py index 503825ec..224ddb95 100644 --- a/GPT_SoVITS/module/mel_processing.py +++ b/GPT_SoVITS/module/mel_processing.py @@ -78,8 +78,9 @@ 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.view_as_real(spec) spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6) return spec @@ -132,6 +133,7 @@ def mel_spectrogram_torch( ) y = y.squeeze(1) + spec = torch.stft( y, n_fft, @@ -142,8 +144,9 @@ def mel_spectrogram_torch( pad_mode="reflect", normalized=False, onesided=True, - return_complex=False, + return_complex=True, ) + spec = torch.view_as_real(spec) spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)