Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix warning

This commit is contained in:
Eren Gölge 2023-06-05 10:29:19 +02:00
parent 724360d71a
commit f26b135e34
3 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/vits.py
View File

@ -131,9 +131,8 @@ def wav_to_spec(y, n_fft, hop_length, win_length, center=False):
pad_mode="reflect",
normalized=False,
onesided=True,
return_complex=False,
)
spec = torch.view_as_real(spec)
spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)
return spec
@ -199,7 +198,6 @@ def wav_to_mel(y, n_fft, num_mels, sample_rate, hop_length, win_length, fmin, fm
pad_mode="reflect",
normalized=False,
onesided=True,
return_complex=False,
)
spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)

2
TTS/utils/audio/torch_transforms.py
View File

@ -129,8 +129,8 @@ class TorchSTFT(nn.Module): # pylint: disable=abstract-method
pad_mode="reflect", # compatible with audio.py
normalized=self.normalized,
onesided=True,
return_complex=False,
)
o = torch.view_as_real(o)
M = o[:, :, :, 0]
P = o[:, :, :, 1]
S = torch.sqrt(torch.clamp(M**2 + P**2, min=1e-8))

6
TTS/vc/modules/freevc/mel_processing.py
View File

@ -64,9 +64,8 @@ def spectrogram_torch(y, n_fft, sampling_rate, hop_size, win_size, center=False)
pad_mode="reflect",
normalized=False,
onesided=True,
return_complex=False,
)
spec = torch.view_as_real(spec)
spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)
return spec
@ -114,9 +113,8 @@ def mel_spectrogram_torch(y, n_fft, num_mels, sampling_rate, hop_size, win_size,
pad_mode="reflect",
normalized=False,
onesided=True,
return_complex=False,
)
spec = torch.view_as_real(spec)
spec = torch.sqrt(spec.pow(2).sum(-1) + 1e-6)
spec = torch.matmul(mel_basis[fmax_dtype_device], spec)