Late-import main TTS libraries in tts CLI

coqui-ai · Sep 26, 2023 · 0a82f06 · 0a82f06
1 parent 5c047cf
commit 0a82f06
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/TTS/bin/synthesize.py b/TTS/bin/synthesize.py
@@ -8,10 +8,6 @@
 # pylint: disable=redefined-outer-name, unused-argument
 from pathlib import Path
 
-from TTS.api import TTS
-from TTS.utils.manage import ModelManager
-from TTS.utils.synthesizer import Synthesizer
-
 description = """
 Synthesize speech on command line.
 
@@ -339,6 +335,11 @@ def main():
     if not any(check_args):
         parser.parse_args(["-h"])
 
+    # Late-import to make things load faster
+    from TTS.api import TTS
+    from TTS.utils.manage import ModelManager
+    from TTS.utils.synthesizer import Synthesizer
+
     # load model manager
     path = Path(__file__).parent / "../.models.json"
     manager = ModelManager(path, progress_bar=args.progress_bar)