From 40cb8ac9660727c4a96022a20810d2bfcde0ae9b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Wed, 1 Dec 2021 10:33:34 +0000
Subject: [PATCH] Fix #958

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

diff --git a/TTS/vocoder/models/gan.py b/TTS/vocoder/models/gan.py
index 81ba87c4..1b49831e 100644
--- a/TTS/vocoder/models/gan.py
+++ b/TTS/vocoder/models/gan.py
@@ -310,6 +310,7 @@ class GAN(BaseVocoder):
         data_items: List,
         verbose: bool,
         num_gpus: int,
+        rank:int=0
     ):
         """Initiate and return the GAN dataloader.