From 353f913efc500bdca73b2b77f57bc0c5d8ef1dca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Wed, 24 Nov 2021 12:48:16 +0100
Subject: [PATCH] Fix #985

---
 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 76fee505..e9bab982 100644
--- a/TTS/vocoder/models/gan.py
+++ b/TTS/vocoder/models/gan.py
@@ -325,6 +325,7 @@ class GAN(BaseVocoder):
             data_items (List): Data samples.
             verbose (bool): Log information if true.
             num_gpus (int): Number of GPUs in use.
+            rank (int): Rank of the current GPU. Defaults to None.
 
         Returns:
             DataLoader: Torch dataloader.