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

glow-tts bug fix

This commit is contained in:
erogol 2021-01-07 13:40:32 +01:00
parent b78aa5d322
commit de2a542f83
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/tts/layers/glow_tts/glow.py
View File

@ -27,7 +27,7 @@ class ConvLayerNorm(nn.Module):
self.conv_layers = nn.ModuleList() self.conv_layers = nn.ModuleList()
self.norm_layers = nn.ModuleList() self.norm_layers = nn.ModuleList()
for idx in range(num_layers - 1): for idx in range(num_layers):
self.conv_layers.append( self.conv_layers.append(
nn.Conv1d(in_channels if idx == 0 else hidden_channels, nn.Conv1d(in_channels if idx == 0 else hidden_channels,
hidden_channels, hidden_channels,

2
TTS/tts/utils/generic_utils.py
View File

@ -109,6 +109,7 @@ def setup_model(num_chars, num_speakers, c, speaker_embedding_dim=None):
out_channels=c.audio['num_mels'], out_channels=c.audio['num_mels'],
encoder_type=c.encoder_type, encoder_type=c.encoder_type,
encoder_params=c.encoder_params, encoder_params=c.encoder_params,
use_encoder_prenet=c["use_encoder_prenet"],
num_flow_blocks_dec=12, num_flow_blocks_dec=12,
kernel_size_dec=5, kernel_size_dec=5,
dilation_rate=1, dilation_rate=1,
@ -120,7 +121,6 @@ def setup_model(num_chars, num_speakers, c, speaker_embedding_dim=None):
num_squeeze=2, num_squeeze=2,
sigmoid_scale=False, sigmoid_scale=False,
mean_only=True, mean_only=True,
use_encoder_prenet=c["use_encoder_prenet"],
external_speaker_embedding_dim=speaker_embedding_dim) external_speaker_embedding_dim=speaker_embedding_dim)
elif c.model.lower() == "speedy_speech": elif c.model.lower() == "speedy_speech":
model = MyModel(num_chars=num_chars + getattr(c, "add_blank", False), model = MyModel(num_chars=num_chars + getattr(c, "add_blank", False),