🔧 Fix #106

TensorSpeech · Jul 9, 2020 · 84c0041 · 84c0041
1 parent a9a48a7
commit 84c0041
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensorflow_tts/models/tacotron2.py b/tensorflow_tts/models/tacotron2.py
@@ -907,6 +907,10 @@ def inference(self, input_ids, input_lengths, speaker_ids):
 
         mel_outputs = decoder_output + residual_projection
 
+        alignment_history = tf.transpose(
+            final_decoder_state.alignment_history.stack(), [1, 2, 0]
+        )
+
         return decoder_output, mel_outputs, stop_token_prediction, alignment_history
 
     @tf.function(