Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

check for speaker id is None before put on cuda

This commit is contained in:
Thomas Werkmeister 2019-07-17 14:08:53 +02:00 committed by GitHub
parent 1468db0d07
commit ee4d55549d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

2
utils/synthesis.py
View File

@ -108,7 +108,7 @@ def synthesis(model,
# preprocess the given text # preprocess the given text
inputs = text_to_seqvec(text, CONFIG, use_cuda) inputs = text_to_seqvec(text, CONFIG, use_cuda)
speaker_id = id_to_torch(speaker_id) speaker_id = id_to_torch(speaker_id)
if use_cuda: if speaker_id is not None and use_cuda:
speaker_id = speaker_id.cuda() speaker_id = speaker_id.cuda()
# synthesize voice # synthesize voice
decoder_output, postnet_output, alignments, stop_tokens = run_model( decoder_output, postnet_output, alignments, stop_tokens = run_model(