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 travis + pylint tests

This commit is contained in:
sanjaesc 2020-10-22 13:22:50 +02:00
parent 6245dd2b93
commit 4d5da4b663
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.travis/script
View File

@ -17,5 +17,6 @@ fi
if [[ "$TEST_SUITE" == "testscripts" ]]; then if [[ "$TEST_SUITE" == "testscripts" ]]; then
# test model training scripts # test model training scripts
./tests/test_tts_train.sh ./tests/test_tts_train.sh
./tests/test_vocoder_train.sh ./tests/test_vocoder_gan_train.sh
./tests/test_vocoder_wavernn_train.sh
fi fi

2
TTS/vocoder/models/wavernn.py
View File

@ -225,7 +225,7 @@ class WaveRNN(nn.Module):
def forward(self, x, mels): def forward(self, x, mels):
bsize = x.size(0) bsize = x.size(0)
h1 = torch.zeros(1, bsize, self.rnn_dims).to(x.device) h1 = torch.zeros(1, bsize, self.rnn_dims).to(x.device)
h2 = torch.zeros(1, bsize, self.rnn_dims).to(x.device) h2 = torch.zeros(1, bsize, self.rnn_dims).to(x.device)
mels, aux = self.upsample(mels) mels, aux = self.upsample(mels)