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

small change for multispeaker

Browse Source 
just threads speaker_id through decoder.run_model
This commit is contained in:
mittimithai
2020-05-12 15:02:24 -07:00
committed by GitHub
parent 7292d303b9
commit 85a822e319
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/synthesizer.py
+7 -2
View File
@@ -164,16 +164,21 @@ class Synthesizer(object):
sentences = list(filter(None, [s.strip() for s in sentences])) # remove empty sentences
return sentences
def tts(self, text):
def tts(self, text, speaker_id=None):
wavs = []
sens = self.split_into_sentences(text)
print(sens)
speaker_id = id_to_torch(speaker_id)
+ if speaker_id is not None and self.use_cuda:
+ speaker_id = speaker_id.cuda()
for sen in sens:
# preprocess the given text
inputs = text_to_seqvec(sen, self.tts_config, self.use_cuda)
# synthesize voice
decoder_output, postnet_output, alignments, _ = run_model(
self.tts_model, inputs, self.tts_config, False, None, None)
self.tts_model, inputs, self.tts_config, False, speaker_id, None)
# convert outputs to numpy
postnet_output, decoder_output, _ = parse_outputs(
postnet_output, decoder_output, alignments)