From d9612a4351cdc5075310452f5a391f068a03af13 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Fri, 16 Apr 2021 18:29:35 +0200 Subject: [PATCH] set the default layer size compatible with scglow --- TTS/tts/models/glow_tts.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/TTS/tts/models/glow_tts.py b/TTS/tts/models/glow_tts.py index 0717e2a8..5c23dd9c 100644 --- a/TTS/tts/models/glow_tts.py +++ b/TTS/tts/models/glow_tts.py @@ -61,6 +61,7 @@ class GlowTTS(nn.Module): external_speaker_embedding_dim=None, ): + super().__init__() self.num_chars = num_chars self.hidden_channels_dp = hidden_channels_dp @@ -88,7 +89,8 @@ class GlowTTS(nn.Module): # if is a multispeaker and c_in_channels is 0, set to 256 if num_speakers > 1: if self.c_in_channels == 0 and not self.external_speaker_embedding_dim: - self.c_in_channels = 512 + # TODO: make this adjustable + self.c_in_channels = 256 elif self.external_speaker_embedding_dim: self.c_in_channels = self.external_speaker_embedding_dim