Fix return outputs

TTS/tts/models/vits.py

@@ -1078,7 +1078,7 @@ class Vits(BaseTTS):
                     scores_disc_real,
                     scores_disc_fake,
                 )
-            return {}, loss_dict
+            return outputs, loss_dict
 
         if optimizer_idx == 1:
             mel = batch["mel"]

tests/tts_tests/test_vits.py

@@ -410,7 +410,9 @@ class TestVits(unittest.TestCase):
             for _ in range(5):
                 batch = self._create_batch(config, 2)
                 for idx in [0, 1]:
-                    _, loss_dict = model.train_step(batch, criterions, idx)
+                    outputs, loss_dict = model.train_step(batch, criterions, idx)
+                    self.assertFalse(not outputs)
+                    self.assertFalse(not loss_dict)
                     loss_dict["loss"].backward()
                     optimizers[idx].step()
                     optimizers[idx].zero_grad()