mirror of https://github.com/coqui-ai/TTS.git
bug fix
This commit is contained in:
parent
31cc9c5443
commit
8ed9f57a6d
|
@ -303,7 +303,7 @@ class Decoder(nn.Module):
|
||||||
stop_token = self.stopnet(output)
|
stop_token = self.stopnet(output)
|
||||||
outputs += [output]
|
outputs += [output]
|
||||||
alignments += [alignment]
|
alignments += [alignment]
|
||||||
stop_tokens += stop_token
|
stop_tokens += [stop_token]
|
||||||
t += 1
|
t += 1
|
||||||
if (not greedy and self.training) or (greedy and memory is not None):
|
if (not greedy and self.training) or (greedy and memory is not None):
|
||||||
if t >= T_decoder:
|
if t >= T_decoder:
|
||||||
|
|
Loading…
Reference in New Issue