diff --git a/models/tacotrongst.py b/models/tacotrongst.py index 5b372338..c18a5e98 100644 --- a/models/tacotrongst.py +++ b/models/tacotrongst.py @@ -38,9 +38,8 @@ class TacotronGST(nn.Module): forward_attn, trans_agent, forward_attn_mask, location_attn, separate_stopnet) self.postnet = PostCBHG(mel_dim) - self.last_linear = nn.Sequential( - nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim), - nn.Sigmoid()) + self.last_linear = nn.Linear(self.postnet.cbhg.gru_features * 2, linear_dim) + def forward(self, characters, text_lengths, mel_specs, speaker_ids=None): B = characters.size(0) diff --git a/setup.py b/setup.py index e9fcc574..ad24ecb1 100644 --- a/setup.py +++ b/setup.py @@ -88,8 +88,6 @@ setup( # "lws", "tqdm", "soundfile", - ], - dependency_links=[ - 'http://github.com/bootphon/phonemizer/tarball/master#egg=phonemizer' + "phonemizer @ https://github.com/bootphon/phonemizer/tarball/master", ], )