✋ Add hifigan to autoconfig and TFautomodel.

Author: dathudeptrai

tensorflow_tts/inference/auto_config.py

@@ -23,6 +23,7 @@
     FastSpeech2Config,
     MelGANGeneratorConfig,
     MultiBandMelGANGeneratorConfig,
+    HifiGANGeneratorConfig,
     Tacotron2Config,
     ParallelWaveGANGeneratorConfig,
 )
@@ -33,6 +34,7 @@
         ("fastspeech2", FastSpeech2Config),
         ("multiband_melgan_generator", MultiBandMelGANGeneratorConfig),
         ("melgan_generator", MelGANGeneratorConfig),
+        ("hifigan_generator", HifiGANGeneratorConfig),
         ("tacotron2", Tacotron2Config),
         ("parallel_wavegan_generator", ParallelWaveGANGeneratorConfig)
     ]

tensorflow_tts/inference/auto_model.py

@@ -23,6 +23,7 @@
     FastSpeech2Config,
     MelGANGeneratorConfig,
     MultiBandMelGANGeneratorConfig,
+    HifiGANGeneratorConfig,
     Tacotron2Config,
     ParallelWaveGANGeneratorConfig,
 )
@@ -32,18 +33,20 @@
     TFFastSpeech2,
     TFMelGANGenerator,
     TFMBMelGANGenerator,
+    TFHifiGANGenerator,
     TFTacotron2,
     TFParallelWaveGANGenerator,
 )
 
 
 TF_MODEL_MAPPING = OrderedDict(
     [
-        (FastSpeechConfig, TFFastSpeech),
         (FastSpeech2Config, TFFastSpeech2),
+        (FastSpeechConfig, TFFastSpeech),
         (MultiBandMelGANGeneratorConfig, TFMBMelGANGenerator),
         (MelGANGeneratorConfig, TFMelGANGenerator),
         (Tacotron2Config, TFTacotron2),
+        (HifiGANGeneratorConfig, TFHifiGANGenerator),
         (ParallelWaveGANGeneratorConfig, TFParallelWaveGANGenerator)
     ]
 )