Fix #958

2021-12-01 10:33:34 +00:00 · 2021-12-01 10:33:34 +00:00 · 40cb8ac966
parent 512ada7548
commit 40cb8ac966
1 changed files with 1 additions and 0 deletions
--- 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.