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

set the default layer size compatible with scglow

This commit is contained in:
Eren Gölge 2021-04-16 18:29:35 +02:00
parent 3ace2440fa
commit c955a12428
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/glow_tts.py
View File

@ -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