check for speaker id is None before put on cuda

2019-07-17 14:08:53 +02:00 · 2019-07-17 14:08:53 +02:00 · ee4d55549d
parent 1468db0d07
commit ee4d55549d
1 changed files with 1 additions and 1 deletions
--- a/utils/synthesis.py
+++ b/utils/synthesis.py
@ -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(