From 8aa6a0decb588c45d6083094371a3c1c019ee347 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 11 Feb 2021 15:28:07 +0000 Subject: [PATCH] set an output_sample_rate in synthesizer and use it for writing the wav file --- TTS/utils/synthesizer.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/TTS/utils/synthesizer.py b/TTS/utils/synthesizer.py index 85e116cf..2a779e53 100644 --- a/TTS/utils/synthesizer.py +++ b/TTS/utils/synthesizer.py @@ -45,8 +45,10 @@ class Synthesizer(object): assert torch.cuda.is_available(), "CUDA is not availabe on this machine." self.load_tts(tts_checkpoint, tts_config, use_cuda) + self.output_sample_rate = self.tts_config.audio['sample_rate'] if vocoder_checkpoint: self.load_vocoder(vocoder_checkpoint, vocoder_config, use_cuda) + self.output_sample_rate = self.vocoder_config.audio['sample_rate'] @staticmethod def get_segmenter(lang): @@ -75,6 +77,7 @@ class Synthesizer(object): def load_tts(self, tts_checkpoint, tts_config, use_cuda): # pylint: disable=global-statement + global symbols, phonemes self.tts_config = load_config(tts_config) @@ -104,7 +107,7 @@ class Synthesizer(object): def save_wav(self, wav, path): wav = np.array(wav) - self.ap.save_wav(wav, path) + self.ap.save_wav(wav, path, self.output_sample_rate) def split_into_sentences(self, text): return self.seg.segment(text)