From d214ac1405f7574a18fe021d169e895828dc6d86 Mon Sep 17 00:00:00 2001
From: Jindrich Matousek <jindrich.matousek@gmail.com>
Date: Tue, 28 Jun 2022 15:22:04 +0200
Subject: [PATCH] fix outputs[0] coming as None proposed by manmay-nakhashi in
 https://github.com/coqui-ai/TTS/pull/1641

---
 TTS/vocoder/models/gan.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TTS/vocoder/models/gan.py b/TTS/vocoder/models/gan.py
index ed5b26dd..1bca9bdf 100644
--- a/TTS/vocoder/models/gan.py
+++ b/TTS/vocoder/models/gan.py
@@ -198,7 +198,7 @@ class GAN(BaseVocoder):
         Returns:
             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"]
         figures = plot_results(y_hat, y, ap, name)
         sample_voice = y_hat[0].squeeze(0).detach().cpu().numpy()