mirror of https://github.com/coqui-ai/TTS.git
call truncated inference
This commit is contained in:
parent
60b6ec18fe
commit
0e0d0345cd
|
@ -33,6 +33,10 @@ def run_model(model, inputs, CONFIG, truncated, speaker_id=None, style_mel=None)
|
|||
if CONFIG.use_gst:
|
||||
decoder_output, postnet_output, alignments, stop_tokens = model.inference(
|
||||
inputs, style_mel=style_mel, speaker_ids=speaker_id)
|
||||
else:
|
||||
if truncated:
|
||||
decoder_output, postnet_output, alignments, stop_tokens = model.inference_truncated(
|
||||
inputs, speaker_ids=speaker_id)
|
||||
else:
|
||||
decoder_output, postnet_output, alignments, stop_tokens = model.inference(
|
||||
inputs, speaker_ids=speaker_id)
|
||||
|
|
Loading…
Reference in New Issue