From d62cac7252eceac2cf49a7ab55c391b85769e30c Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Mon, 28 Dec 2020 13:51:15 +0100
Subject: [PATCH] fix glow-tts prenet bug fix

---
 TTS/tts/layers/glow_tts/encoder.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/TTS/tts/layers/glow_tts/encoder.py b/TTS/tts/layers/glow_tts/encoder.py
index ef4dc28b..b57370fe 100644
--- a/TTS/tts/layers/glow_tts/encoder.py
+++ b/TTS/tts/layers/glow_tts/encoder.py
@@ -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