control synthesis lenght as an additional stop condition

layers/tacotron2.py

@@ -388,7 +388,7 @@ class Decoder(nn.Module):
 
         self.attention_layer.init_win_idx()
         outputs, stop_tokens, alignments, t = [], [], [], 0
-        stop_flags = [False, False]
+        stop_flags = [False, False, False]
         while True:
             memory = self.prenet(memory)
             mel_output, stop_token, alignment = self.decode(memory)
@@ -398,7 +398,8 @@ class Decoder(nn.Module):
             alignments += [alignment]
 
             stop_flags[0] = stop_flags[0] or stop_token > 0.5
-            stop_flags[1] = stop_flags[1] or alignment[0, -2:].sum() > 0.5
+            stop_flags[1] = stop_flags[1] or (alignment[0, -2:].sum() > 0.5 and t > inputs.shape[1])
+            stop_flags[2] = t > inputs.shape[1]
             if all(stop_flags):
                 break
             elif len(outputs) == self.max_decoder_steps: