From e1b4c4ca43d390a2fe19a38b8a7660721532bc1e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Wed, 24 Nov 2021 12:59:51 +0100 Subject: [PATCH] Add init_from_config to GAN --- TTS/vocoder/models/gan.py | 4 ++++ 1 file changed, 4 insertions(+) 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