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

linter fixes and version updates for deps

This commit is contained in:
Eren Gölge 2021-02-01 13:18:56 +00:00 committed by Eren Gölge
parent a30a231566
commit 93a6bdfd6c
4 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
hubconf.py
View File

@ -1,9 +1,6 @@
dependencies = ['torch', 'gdown'] dependencies = ['torch', 'gdown']
import torch import torch
import os
import zipfile
from TTS.utils.generic_utils import get_user_data_dir
from TTS.utils.synthesizer import Synthesizer from TTS.utils.synthesizer import Synthesizer
from TTS.utils.manage import ModelManager from TTS.utils.manage import ModelManager
@ -30,8 +27,8 @@ def tts(model_name='tts_models/en/ljspeech/tacotron2-DCA', vocoder_name='vocoder
vocoder_path, vocoder_config_path = manager.download_model(vocoder_name) vocoder_path, vocoder_config_path = manager.download_model(vocoder_name)
# create synthesizer # create synthesizer
synthesizer = Synthesizer(model_path, config_path, vocoder_path, vocoder_config_path) synt = Synthesizer(model_path, config_path, vocoder_path, vocoder_config_path)
return synthesizer return synt
if __name__ == '__main__': if __name__ == '__main__':

3
tests/test_vocoder_gan_datasets.py
View File

@ -61,7 +61,8 @@ def gan_dataset_case(batch_size, seq_len, hop_len, conv_pad, return_segments, us
mel = ap.melspectrogram(audio) mel = ap.melspectrogram(audio)
# the first 2 and the last 2 frames are skipped due to the padding # the first 2 and the last 2 frames are skipped due to the padding
# differences in stft # differences in stft
assert (feat - mel[:, :feat1.shape[-1]])[:, 2:-2].sum() <= 0, f' [!] {(feat - mel[:, :feat1.shape[-1]])[:, 2:-2].sum()}' max_diff = abs((feat - mel[:, :feat1.shape[-1]])[:, 2:-2]).max()
assert max_diff <= 0, f' [!] {max_diff}'
count_iter += 1 count_iter += 1
# if count_iter == max_iter: # if count_iter == max_iter: