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

fix outputs[0] coming as None 

proposed by manmay-nakhashi in https://github.com/coqui-ai/TTS/pull/1641
This commit is contained in:
Jindrich Matousek 2022-06-28 15:22:04 +02:00
parent 0972653b06
commit d214ac1405
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/vocoder/models/gan.py
View File

@ -198,7 +198,7 @@ class GAN(BaseVocoder):
Returns: Returns:
Tuple[Dict, Dict]: log figures and audio samples. Tuple[Dict, Dict]: log figures and audio samples.
""" """
y_hat = outputs[0]["model_outputs"] y_hat = outputs[0]["model_outputs"] if outputs[0] is not None else outputs[1]["model_outputs"]
y = batch["waveform"] y = batch["waveform"]
figures = plot_results(y_hat, y, ap, name) figures = plot_results(y_hat, y, ap, name)
sample_voice = y_hat[0].squeeze(0).detach().cpu().numpy() sample_voice = y_hat[0].squeeze(0).detach().cpu().numpy()