diff --git a/TTS/vocoder/models/gan.py b/TTS/vocoder/models/gan.py index e9bab982..b4e3652e 100644 --- a/TTS/vocoder/models/gan.py +++ b/TTS/vocoder/models/gan.py @@ -361,3 +361,7 @@ class GAN(BaseVocoder): def get_criterion(self): """Return criterions for the optimizers""" return [GeneratorLoss(self.config), DiscriminatorLoss(self.config)] + + @staticmethod + def init_from_config(config: Coqpit) -> "GAN": + return GAN(config) \ No newline at end of file