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.