🔧 Fix #106

Author: dathudeptrai

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(