From 5723eb47383fb38b6caadf0f431403e99aeeff42 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 29 Jun 2021 16:41:08 +0200
Subject: [PATCH] Fix config init in `process_args`

---
 TTS/trainer.py                            | 21 +++++++++++----------
 recipes/ljspeech/hifigan/train_hifigan.py |  3 +--
 2 files changed, 12 insertions(+), 12 deletions(-)

diff --git a/TTS/trainer.py b/TTS/trainer.py
index b28eceb5..93efeef4 100644
--- a/TTS/trainer.py
+++ b/TTS/trainer.py
@@ -1012,17 +1012,18 @@ def process_args(args, config=None):
         args.restore_path, best_model = get_last_checkpoint(args.continue_path)
         if not args.best_path:
             args.best_path = best_model
-    # init config
-    if config is None and args.config_path:
-        # init from a file
-        config = load_config(args.config_path)
-    else:
-        # init from console args
-        from TTS.config.shared_configs import BaseTrainingConfig  # pylint: disable=import-outside-toplevel
+    # init config if not already defined
+    if config is None:
+        if args.config_path:
+            # init from a file
+            config = load_config(args.config_path)
+        else:
+            # init from console args
+            from TTS.config.shared_configs import BaseTrainingConfig  # pylint: disable=import-outside-toplevel
 
-        config_base = BaseTrainingConfig()
-        config_base.parse_known_args(coqpit_overrides)
-        config = register_config(config_base.model)()
+            config_base = BaseTrainingConfig()
+            config_base.parse_known_args(coqpit_overrides)
+            config = register_config(config_base.model)()
     # override values from command-line args
     config.parse_known_args(coqpit_overrides, relaxed_parser=True)
     if config.mixed_precision:
diff --git a/recipes/ljspeech/hifigan/train_hifigan.py b/recipes/ljspeech/hifigan/train_hifigan.py
index 99b39e99..af615ace 100644
--- a/recipes/ljspeech/hifigan/train_hifigan.py
+++ b/recipes/ljspeech/hifigan/train_hifigan.py
@@ -1,8 +1,7 @@
 import os
 
+from TTS.trainer import Trainer, TrainingArgs, init_training
 from TTS.vocoder.configs import HifiganConfig
-from TTS.trainer import init_training, Trainer, TrainingArgs
-
 
 output_path = os.path.dirname(os.path.abspath(__file__))
 config = HifiganConfig(