fix constant GL bug in synthesis

This commit is contained in:
erogol 2020-02-19 17:54:06 +01:00
parent df42a4a03a
commit 6977899d07
1 changed files with 7 additions and 4 deletions

View File

@ -78,6 +78,7 @@ def synthesis(model,
style_wav=None, style_wav=None,
truncated=False, truncated=False,
enable_eos_bos_chars=False, #pylint: disable=unused-argument enable_eos_bos_chars=False, #pylint: disable=unused-argument
use_griffin_lim=False,
do_trim_silence=False): do_trim_silence=False):
"""Synthesize voice for the given text. """Synthesize voice for the given text.
@ -111,8 +112,10 @@ def synthesis(model,
postnet_output, decoder_output, alignment = parse_outputs( postnet_output, decoder_output, alignment = parse_outputs(
postnet_output, decoder_output, alignments) postnet_output, decoder_output, alignments)
# plot results # plot results
wav = inv_spectrogram(postnet_output, ap, CONFIG) wav = None
# trim silence if use_griffin_lim:
if do_trim_silence: wav = inv_spectrogram(postnet_output, ap, CONFIG)
wav = trim_silence(wav, ap) # trim silence
if do_trim_silence:
wav = trim_silence(wav, ap)
return wav, alignment, decoder_output, postnet_output, stop_tokens return wav, alignment, decoder_output, postnet_output, stop_tokens