wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix glow-tts prenet bug fix

Browse Source
This commit is contained in:
erogol
2021-01-06 13:19:40 +01:00
parent a1d5a9ddda
commit d62cac7252
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/layers/glow_tts/encoder.py
+2 -3
View File
@@ -136,9 +136,8 @@ class Encoder(nn.Module):
x_mask = torch.unsqueeze(sequence_mask(x_lengths, x.size(2)),
1).to(x.dtype)
# pre-conv layers
if self.encoder_type in ['transformer', 'time-depth-separable']:
if self.use_prenet:
x = self.pre(x, x_mask)
if hasattr(self, 'pre') and self.use_prenet:
x = self.pre(x, x_mask)
# encoder
x = self.encoder(x, x_mask)
# set duration predictor input