mirror of
https://github.com/wassname/TTS.git
synced 2026-10-24 12:00:20 +08:00
check for speaker id is None before put on cuda
This commit is contained in:
1 parent
1468db0d07
commit
ee4d55549d
1 file changed
+1
-1
+1
-1
@@ -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(
|
||||
|
||||
Reference in new issue
Block a user