diff --git a/mozilla_voice_tts/tts/layers/glow_tts/encoder.py b/mozilla_voice_tts/tts/layers/glow_tts/encoder.py index 1a1fe2ca..d8bbc07c 100644 --- a/mozilla_voice_tts/tts/layers/glow_tts/encoder.py +++ b/mozilla_voice_tts/tts/layers/glow_tts/encoder.py @@ -108,7 +108,7 @@ class Encoder(nn.Module): # pass embedding layer # [B ,T, D] x = self.emb(x) * math.sqrt(self.hidden_channels) - x += self.pe[:x.shape[1]].unsqueeze(0) + x = x + self.pe[:x.shape[1]].unsqueeze(0) # [B, D, T] x = torch.transpose(x, 1, -1) # compute input sequence mask