From bb10b87e98b191524a79618b52140c33337fb52f Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Tue, 27 Aug 2024 15:21:31 +0200
Subject: [PATCH] improve logging

---
 LLM/language_model.py                | 3 ---
 LLM/mlx_language_model.py            | 3 ---
 STT/lightning_whisper_mlx_handler.py | 3 ---
 STT/whisper_stt_handler.py           | 4 ----
 TTS/melo_handler.py                  | 3 ---
 TTS/parler_handler.py                | 3 ---
 VAD/vad_handler.py                   | 3 ---
 connections/socket_receiver.py       | 3 ---
 connections/socket_sender.py         | 3 ---
 s2s_pipeline.py                      | 2 +-
 10 files changed, 1 insertion(+), 29 deletions(-)

diff --git a/LLM/language_model.py b/LLM/language_model.py
index 372d21d..bc39b23 100644
--- a/LLM/language_model.py
+++ b/LLM/language_model.py
@@ -13,9 +13,6 @@ from rich.console import Console
 import logging
 from nltk import sent_tokenize
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/LLM/mlx_language_model.py b/LLM/mlx_language_model.py
index 65c3b2d..7e041de 100644
--- a/LLM/mlx_language_model.py
+++ b/LLM/mlx_language_model.py
@@ -5,9 +5,6 @@ from mlx_lm import load, stream_generate, generate
 from rich.console import Console
 import torch
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/STT/lightning_whisper_mlx_handler.py b/STT/lightning_whisper_mlx_handler.py
index be770ac..e5cd540 100644
--- a/STT/lightning_whisper_mlx_handler.py
+++ b/STT/lightning_whisper_mlx_handler.py
@@ -6,9 +6,6 @@ import numpy as np
 from rich.console import Console
 import torch
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py
index fc87bd3..ea37b8b 100644
--- a/STT/whisper_stt_handler.py
+++ b/STT/whisper_stt_handler.py
@@ -9,11 +9,7 @@ from baseHandler import BaseHandler
 from rich.console import Console
 import logging
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
-
 console = Console()
 
 
diff --git a/TTS/melo_handler.py b/TTS/melo_handler.py
index 9897679..376747b 100644
--- a/TTS/melo_handler.py
+++ b/TTS/melo_handler.py
@@ -6,9 +6,6 @@ import numpy as np
 from rich.console import Console
 import torch
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/TTS/parler_handler.py b/TTS/parler_handler.py
index 06dc19b..5cc0ce9 100644
--- a/TTS/parler_handler.py
+++ b/TTS/parler_handler.py
@@ -19,9 +19,6 @@ torch._inductor.config.fx_graph_cache = True
 # mind about this parameter ! should be >= 2 * number of padded prompt sizes for TTS
 torch._dynamo.config.cache_size_limit = 15
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/VAD/vad_handler.py b/VAD/vad_handler.py
index e2da03f..3a36498 100644
--- a/VAD/vad_handler.py
+++ b/VAD/vad_handler.py
@@ -8,9 +8,6 @@ from utils.utils import int2float
 
 import logging
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/connections/socket_receiver.py b/connections/socket_receiver.py
index 19bda8d..6fb734c 100644
--- a/connections/socket_receiver.py
+++ b/connections/socket_receiver.py
@@ -2,9 +2,6 @@ import socket
 from rich.console import Console
 import logging
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/connections/socket_sender.py b/connections/socket_sender.py
index 587343b..11ed210 100644
--- a/connections/socket_sender.py
+++ b/connections/socket_sender.py
@@ -2,9 +2,6 @@ import socket
 from rich.console import Console
 import logging
 
-logging.basicConfig(
-    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-)
 logger = logging.getLogger(__name__)
 
 console = Console()
diff --git a/s2s_pipeline.py b/s2s_pipeline.py
index 7a9e204..c5d8e13 100644
--- a/s2s_pipeline.py
+++ b/s2s_pipeline.py
@@ -43,8 +43,8 @@ except (LookupError, OSError):
 CURRENT_DIR = Path(__file__).resolve().parent
 os.environ["TORCHINDUCTOR_CACHE_DIR"] = os.path.join(CURRENT_DIR, "tmp")
 
-
 console = Console()
+logging.getLogger("numba").setLevel(logging.WARNING)  # quiet down numba logs
 
 
 def prepare_args(args, prefix):
-- 
GitLab