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

Replace values by dict in output of synthesis() in test_run() in vits

This commit is contained in:
Jindrich Matousek 2023-02-25 11:26:55 +01:00
parent abca11714e
commit 8a29d57ff0
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/tts/models/vits.py
View File

@ -1439,7 +1439,8 @@ class Vits(BaseTTS):
test_sentences = self.config.test_sentences
for idx, s_info in enumerate(test_sentences):
aux_inputs = self.get_aux_input_from_test_sentences(s_info)
wav, alignment, _, _ = synthesis(
# JMa: replace individual variables with dictionary
outputs = synthesis(
self,
aux_inputs["text"],
self.config,
@ -1450,9 +1451,9 @@ class Vits(BaseTTS):
language_id=aux_inputs["language_id"],
use_griffin_lim=True,
do_trim_silence=False,
).values()
test_audios["{}-audio".format(idx)] = wav
test_figures["{}-alignment".format(idx)] = plot_alignment(alignment.T, output_fig=False)
)
test_audios["{}-audio".format(idx)] = outputs["wav"]
test_figures["{}-alignment".format(idx)] = plot_alignment(outputs["alignments"].T, output_fig=False)
return {"figures": test_figures, "audios": test_audios}
def test_log(