From 70bd84894db3736836af8f9503728528fd71706c Mon Sep 17 00:00:00 2001 From: Enno Hermann Date: Mon, 20 May 2024 11:26:34 +0200 Subject: [PATCH] fix(server): ensure logging output gets actually shown --- TTS/encoder/utils/prepare_voxceleb.py | 4 +++- TTS/server/server.py | 3 ++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/TTS/encoder/utils/prepare_voxceleb.py b/TTS/encoder/utils/prepare_voxceleb.py index 8f571dd2..da7522a5 100644 --- a/TTS/encoder/utils/prepare_voxceleb.py +++ b/TTS/encoder/utils/prepare_voxceleb.py @@ -29,6 +29,8 @@ import zipfile import soundfile as sf +from TTS.utils.generic_utils import ConsoleFormatter, setup_logger + logger = logging.getLogger(__name__) SUBSETS = { @@ -214,7 +216,7 @@ def processor(directory, subset, force_process): if __name__ == "__main__": - logging.getLogger("TTS").setLevel(logging.INFO) + setup_logger("TTS", level=logging.INFO, screen=True, formatter=ConsoleFormatter()) if len(sys.argv) != 4: print("Usage: python prepare_data.py save_directory user password") sys.exit() diff --git a/TTS/server/server.py b/TTS/server/server.py index a8f3a088..54c1ad45 100644 --- a/TTS/server/server.py +++ b/TTS/server/server.py @@ -16,11 +16,12 @@ except ImportError as e: raise ImportError("Server requires requires flask, use `pip install coqui-tts[server]`.") from e from TTS.config import load_config +from TTS.utils.generic_utils import ConsoleFormatter, setup_logger from TTS.utils.manage import ModelManager from TTS.utils.synthesizer import Synthesizer logger = logging.getLogger(__name__) -logging.getLogger("TTS").setLevel(logging.INFO) +setup_logger("TTS", level=logging.INFO, screen=True, formatter=ConsoleFormatter()) def create_argparser():