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)
|
||||
outputs += [output]
|
||||
alignments += [alignment]
|
||||
stop_tokens += stop_token
|
||||
stop_tokens += [stop_token]
|
||||
t += 1
|
||||
if (not greedy and self.training) or (greedy and memory is not None):
|
||||
if t >= T_decoder:
|
||||
|
|
Loading…
Reference in New Issue