From 4cd0e4eb0d3dbd4621b9b5125aeb4c2ba6427bf5 Mon Sep 17 00:00:00 2001 From: WeberJulian Date: Tue, 23 Nov 2021 15:40:03 +0100 Subject: [PATCH] Remove self.audio_config from VITS --- TTS/tts/models/vits.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index 4eb12b3b..6acf2000 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -270,7 +270,6 @@ class Vits(BaseTTS): super().__init__(config) self.END2END = True - self.speaker_manager = speaker_manager self.audio_config = config["audio"] if config.__class__.__name__ == "VitsConfig": @@ -408,7 +407,7 @@ class Vits(BaseTTS): if ( hasattr(self.speaker_encoder, "audio_config") - and self.audio_config["sample_rate"] != self.speaker_encoder.audio_config["sample_rate"] + and self.config.audio["sample_rate"] != self.speaker_encoder.audio_config["sample_rate"] ): raise RuntimeError( " [!] To use the speaker consistency loss (SCL) you need to have the TTS model sampling rate ({}) equal to the speaker encoder sampling rate ({}) !".format(self.audio_config["sample_rate"], self.speaker_encoder.audio_config["sample_rate"])