coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set an output_sample_rate in synthesizer and use it for writing the wav 

file
pull/367/head
Eren Gölge 2021-02-11 15:28:07 +00:00 committed by Eren Gölge
parent 3ccb015cd8
commit d30608ab17
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/utils/synthesizer.py
View File

@ -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)