From 3c160131996224569ddc68678fcc98f9f1d90f14 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Fri, 10 Sep 2021 08:26:34 +0000 Subject: [PATCH] Fix Vits imports --- TTS/tts/models/vits.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 52f34e6e..87695774 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -9,12 +9,11 @@ from torch import nn from torch.cuda.amp.autocast_mode import autocast from TTS.tts.layers.glow_tts.duration_predictor import DurationPredictor -from TTS.tts.utils.helpers import generate_path, maximum_path from TTS.tts.layers.vits.discriminator import VitsDiscriminator from TTS.tts.layers.vits.networks import PosteriorEncoder, ResidualCouplingBlocks, TextEncoder from TTS.tts.layers.vits.stochastic_duration_predictor import StochasticDurationPredictor from TTS.tts.models.base_tts import BaseTTS -from TTS.tts.utils.helpers import sequence_mask +from TTS.tts.utils.helpers import generate_path, maximum_path, rand_segments, segment, sequence_mask from TTS.tts.utils.speakers import get_speaker_manager from TTS.tts.utils.synthesis import synthesis from TTS.tts.utils.visual import plot_alignment @@ -22,7 +21,6 @@ from TTS.utils.audio import AudioProcessor from TTS.utils.trainer_utils import get_optimizer, get_scheduler from TTS.vocoder.models.hifigan_generator import HifiganGenerator from TTS.vocoder.utils.generic_utils import plot_results -from TTS.tts.utils.helpers import rand_segments, segment @dataclass