import missings for tacotron.py

This commit is contained in:
Eren Gölge
2021-06-28 17:03:19 +02:00
parent da7d10e53c
commit 5a2e75f0ee
+2
View File
@@ -2,6 +2,8 @@
import torch
from torch import nn
from TTS.tts.utils.measures import alignment_diagonal_score
from TTS.tts.utils.visual import plot_alignment, plot_spectrogram
from TTS.tts.layers.tacotron.gst_layers import GST
from TTS.tts.layers.tacotron.tacotron import Decoder, Encoder, PostCBHG
from TTS.tts.models.tacotron_abstract import TacotronAbstract