[TTS] minor fix typos and input_types (NVIDIA#7272)

Signed-off-by: Xuesong Yang <[email protected]> Signed-off-by: dorotat <[email protected]>
dorotat-nv · Aug 24, 2023 · a0e7f48 · a0e7f48
1 parent 95df78d
commit a0e7f48
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/nemo/collections/tts/models/hifigan.py b/nemo/collections/tts/models/hifigan.py
@@ -75,6 +75,8 @@ def __init__(self, cfg: DictConfig, trainer: 'Trainer' = None):
         self.log_audio = cfg.get("log_audio", False)
         self.log_config = cfg.get("log_config", None)
         self.lr_schedule_interval = None
+
+        # Important: this property activates manual optimization.
         self.automatic_optimization = False
 
     @property
@@ -350,7 +352,7 @@ def __setup_dataloader_from_config(self, cfg, shuffle_should_be: bool = True, na
         if "dataset" not in cfg or not isinstance(cfg.dataset, DictConfig):
             raise ValueError(f"No dataset for {name}")
         if "dataloader_params" not in cfg or not isinstance(cfg.dataloader_params, DictConfig):
-            raise ValueError(f"No dataloder_params for {name}")
+            raise ValueError(f"No dataloader_params for {name}")
         if shuffle_should_be:
             if 'shuffle' not in cfg.dataloader_params:
                 logging.warning(

diff --git a/nemo/collections/tts/modules/hifigan_modules.py b/nemo/collections/tts/modules/hifigan_modules.py
@@ -334,7 +334,7 @@ def __init__(self, debug=False):
         )
 
     @property
-    def output_types(self):
+    def input_types(self):
         return {
             "y": NeuralType(('B', 'S', 'T'), AudioSignal()),
             "y_hat": NeuralType(('B', 'S', 'T'), AudioSignal()),