wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
file
This commit is contained in:
Eren Gölge
2021-02-11 15:28:07 +00:00
parent 0c52d27d65
commit 8aa6a0decb
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/utils/synthesizer.py
+4 -1
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)