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

Stop test time model with stop_token

This commit is contained in:
Eren Golge 2018-05-13 06:31:59 -07:00
parent bbad1ce28d
commit 5ce80d052c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
layers/tacotron.py
View File

@ -309,7 +309,7 @@ class Decoder(nn.Module):
if t >= T_decoder: if t >= T_decoder:
break break
else: else:
if t > 1 and is_end_of_frames(output.view(self.r, -1), alignment, self.eps): if t > 1 and stop_token > 0.5:
break break
elif t > self.max_decoder_steps: elif t > self.max_decoder_steps:
print(" !! Decoder stopped with 'max_decoder_steps'. \ print(" !! Decoder stopped with 'max_decoder_steps'. \