Fix enhancer usage

Signed-off-by: Vladimir Bataev <[email protected]>
NVIDIA · Jun 30, 2023 · 0894b32 · 0894b32
1 parent 8d1c018
commit 0894b32
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nemo/collections/asr/models/hybrid_asr_tts_models.py b/nemo/collections/asr/models/hybrid_asr_tts_models.py
@@ -311,8 +311,10 @@ def from_pretrained_models(
                 )
             )
         else:
+            cfg = copy.deepcopy(cfg)  # copy to avoid modifying original config
             cfg.tts_model_path = f"{tts_model_path}"
             cfg.asr_model_path = f"{asr_model_path}"
+            cfg.enhancer_model_path = f"{enhancer_model_path}" if enhancer_model_path is not None else None
         return ASRWithTTSModel(cfg, trainer=trainer)
 
     def __setattr__(self, name, value):