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

pass x_mask to layer norm

This commit is contained in:
erogol 2020-08-17 10:58:49 +02:00
parent 53523eebbe
commit 7c2c4d6f27
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/vocoder/layers/pqmf.py
View File

@ -22,7 +22,7 @@ class PQMF(torch.nn.Module):
for k in range(N): for k in range(N):
constant_factor = (2 * k + 1) * (np.pi / constant_factor = (2 * k + 1) * (np.pi /
(2 * N)) * (np.arange(taps + 1) - (2 * N)) * (np.arange(taps + 1) -
((taps - 1) / 2)) ((taps - 1) / 2)) # TODO: (taps - 1) -> taps
phase = (-1)**k * np.pi / 4 phase = (-1)**k * np.pi / 4
H[k] = 2 * QMF * np.cos(constant_factor + phase) H[k] = 2 * QMF * np.cos(constant_factor + phase)

2
mozilla_voice_tts/tts/layers/glow_tts/encoder.py
View File

@ -121,7 +121,7 @@ class Encoder(nn.Module):
1).to(x.dtype) 1).to(x.dtype)
# pass encoder # pass encoder
for layer in self.encoder: for layer in self.encoder:
x = layer(x) x = layer(x, x_mask)
# set duration predictor input # set duration predictor input
if g is not None: if g is not None:
g_exp = g.expand(-1, -1, x.size(-1)) g_exp = g.expand(-1, -1, x.size(-1))