mirror of https://github.com/coqui-ai/TTS.git
make optional position encoding for speedyspeech
This commit is contained in:
parent
dc4a16d62e
commit
5cae2c5742
|
@ -131,6 +131,7 @@ def setup_model(num_chars, num_speakers, c, speaker_embedding_dim=None):
|
||||||
model = MyModel(num_chars=num_chars + getattr(c, "add_blank", False),
|
model = MyModel(num_chars=num_chars + getattr(c, "add_blank", False),
|
||||||
out_channels=c.audio['num_mels'],
|
out_channels=c.audio['num_mels'],
|
||||||
hidden_channels=128,
|
hidden_channels=128,
|
||||||
|
positional_encoding=c['positional_encoding'],
|
||||||
encoder_type=c['encoder_type'],
|
encoder_type=c['encoder_type'],
|
||||||
decoder_residual_conv_bn_params=c['decoder_residual_conv_bn_params'],
|
decoder_residual_conv_bn_params=c['decoder_residual_conv_bn_params'],
|
||||||
c_in_channels=0)
|
c_in_channels=0)
|
||||||
|
|
Loading…
Reference in New Issue