blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 07:02:57 +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",
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)