small fixes

TTS/vocoder/utils/generic_utils.py

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

run_tests.sh

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

tests/inputs/test_align_tts.json

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

tests/test_vocoder_losses.py

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