diff --git a/TTS/utils/arguments.py b/TTS/utils/arguments.py index 6a09986c..ea195767 100644 --- a/TTS/utils/arguments.py +++ b/TTS/utils/arguments.py @@ -176,7 +176,7 @@ def process_args(args, model_type): _ = os.path.dirname(os.path.realpath(__file__)) - if model_type in "tacotron wavegrad wavernn" and c.mixed_precision: + if c.mixed_precision: print(" > Mixed precision mode is ON") out_path = args.continue_path