From 85a822e3190f6c7207c28f7485ecb0ebe7e480ba Mon Sep 17 00:00:00 2001 From: mittimithai Date: Tue, 12 May 2020 15:02:24 -0700 Subject: [PATCH] small change for multispeaker just threads speaker_id through decoder.run_model --- server/synthesizer.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/server/synthesizer.py b/server/synthesizer.py index e9205bf1..411be928 100644 --- a/server/synthesizer.py +++ b/server/synthesizer.py @@ -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)