🦠 Add ParalleWavegan to Auto module.

Author: dathudeptrai

tensorflow_tts/inference/auto_config.py

@@ -24,6 +24,7 @@
     MelGANGeneratorConfig,
     MultiBandMelGANGeneratorConfig,
     Tacotron2Config,
+    ParallelWaveGANGeneratorConfig,
 )
 
 CONFIG_MAPPING = OrderedDict(
@@ -32,7 +33,8 @@
         ("fastspeech2", FastSpeech2Config),
         ("multiband_melgan_generator", MultiBandMelGANGeneratorConfig),
         ("melgan_generator", MelGANGeneratorConfig),
-        ("tacotron2", Tacotron2Config)
+        ("tacotron2", Tacotron2Config),
+        ("parallel_wavegan_generator", ParallelWaveGANGeneratorConfig)
     ]
 )
 

tensorflow_tts/inference/auto_model.py

@@ -24,6 +24,7 @@
     MelGANGeneratorConfig,
     MultiBandMelGANGeneratorConfig,
     Tacotron2Config,
+    ParallelWaveGANGeneratorConfig,
 )
 
 from tensorflow_tts.models import (
@@ -32,6 +33,7 @@
     TFMelGANGenerator,
     TFMBMelGANGenerator,
     TFTacotron2,
+    TFParallelWaveGANGenerator,
 )
 
 
@@ -42,6 +44,7 @@
         (MultiBandMelGANGeneratorConfig, TFMBMelGANGenerator),
         (MelGANGeneratorConfig, TFMelGANGenerator),
         (Tacotron2Config, TFTacotron2),
+        (ParallelWaveGANGeneratorConfig, TFParallelWaveGANGenerator)
     ]
 )
 

test/test_auto.py

@@ -58,6 +58,7 @@ def test_auto_processor(mapper_path):
         "./examples/multiband_melgan/conf/multiband_melgan.v1.yaml",
         "./examples/tacotron2/conf/tacotron2.v1.yaml",
         "./examples/tacotron2/conf/tacotron2.kss.v1.yaml",
+        "./examples/parallel_wavegan/conf/parallel_wavegan.v1.yaml",
      ]
 )
 def test_auto_model(config_path):