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

small fixes

This commit is contained in:
Eren Gölge 2021-04-08 14:22:47 +02:00
parent a7fb49843b
commit 53f54898bc
4 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/vocoder/utils/generic_utils.py
View File

@ -96,7 +96,7 @@ def setup_generator(c):
in_channels=c.audio['num_mels'], in_channels=c.audio['num_mels'],
out_channels=1, out_channels=1,
**c.generator_model_params) **c.generator_model_params)
if c.generator_model.lower() in 'melgan_generator': elif c.generator_model.lower() in 'melgan_generator':
model = MyModel( model = MyModel(
in_channels=c.audio['num_mels'], in_channels=c.audio['num_mels'],
out_channels=1, out_channels=1,

2
run_tests.sh
View File

@ -4,7 +4,7 @@ TF_CPP_MIN_LOG_LEVEL=3
# # tests # # tests
nosetests tests -x &&\ nosetests tests -x &&\
# runtime tests # # runtime tests
./tests/test_demo_server.sh && \ ./tests/test_demo_server.sh && \
./tests/test_resample.sh && \ ./tests/test_resample.sh && \
./tests/test_tacotron_train.sh && \ ./tests/test_tacotron_train.sh && \

1
tests/inputs/test_align_tts.json
View File

@ -65,6 +65,7 @@
// MODEL PARAMETERS // MODEL PARAMETERS
"positional_encoding": true, "positional_encoding": true,
"hidden_channels": 256, "hidden_channels": 256,
"hidden_channels_dp": 128,
"encoder_type": "fftransformer", "encoder_type": "fftransformer",
"encoder_params":{ "encoder_params":{
"hidden_channels_ffn": 1024 , "hidden_channels_ffn": 1024 ,

2
tests/test_vocoder_losses.py
View File

@ -53,6 +53,7 @@ def test_multiscale_stft_loss():
assert loss_sc < 1.0 assert loss_sc < 1.0
assert loss_m + loss_sc > 0 assert loss_m + loss_sc > 0
def test_melgan_feature_loss(): def test_melgan_feature_loss():
feats_real = [] feats_real = []
feats_fake = [] feats_fake = []
@ -71,7 +72,6 @@ def test_melgan_feature_loss():
loss = loss_func(feats_fake, feats_real) loss = loss_func(feats_fake, feats_real)
assert loss.item() <= 1.0 assert loss.item() <= 1.0
feats_real = [] feats_real = []
feats_fake = [] feats_fake = []