Merge pull request #229 from twerkmeister/patch-2

check for speaker id is None before put on cuda
This commit is contained in:
Eren Gölge
2019-07-17 16:20:24 +02:00
committed by GitHub
+1 -1
View File
@@ -108,7 +108,7 @@ def synthesis(model,
# preprocess the given text
inputs = text_to_seqvec(text, CONFIG, use_cuda)
speaker_id = id_to_torch(speaker_id)
if use_cuda:
if speaker_id is not None and use_cuda:
speaker_id = speaker_id.cuda()
# synthesize voice
decoder_output, postnet_output, alignments, stop_tokens = run_model(