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 glow-tts prenet bug fix

This commit is contained in:
erogol 2020-12-28 13:51:15 +01:00
parent a1d5a9ddda
commit d62cac7252
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/tts/layers/glow_tts/encoder.py
View File

@ -136,9 +136,8 @@ class Encoder(nn.Module):
x_mask = torch.unsqueeze(sequence_mask(x_lengths, x.size(2)), x_mask = torch.unsqueeze(sequence_mask(x_lengths, x.size(2)),
1).to(x.dtype) 1).to(x.dtype)
# pre-conv layers # pre-conv layers
if self.encoder_type in ['transformer', 'time-depth-separable']: if hasattr(self, 'pre') and self.use_prenet:
if self.use_prenet: x = self.pre(x, x_mask)
x = self.pre(x, x_mask)
# encoder # encoder
x = self.encoder(x, x_mask) x = self.encoder(x, x_mask)
# set duration predictor input # set duration predictor input