blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 15:19:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove stft warning

Browse Source
This commit is contained in:
SapphireLab 2024-03-06 02:20:30 +08:00
parent 78ab26ea17
commit 7c905bb152
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
GPT_SoVITS/module/mel_processing.py
View File

@ -78,8 +78,9 @@ def spectrogram_torch(y, n_fft, sampling_rate, hop_size, win_size, center=False)
pad_mode="reflect", pad_mode="reflect",
normalized=False, normalized=False,
onesided=True, onesided=True,
return_complex=False, return_complex=True
) )
spec = torch.view_as_real(spec)
spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6) spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)
return spec return spec
@ -132,6 +133,7 @@ def mel_spectrogram_torch(
) )
y = y.squeeze(1) y = y.squeeze(1)
spec = torch.stft( spec = torch.stft(
y, y,
n_fft, n_fft,
@ -142,8 +144,9 @@ def mel_spectrogram_torch(
pad_mode="reflect", pad_mode="reflect",
normalized=False, normalized=False,
onesided=True, onesided=True,
return_complex=False, return_complex=True,
) )
spec = torch.view_as_real(spec)
spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6) spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)