feat(ipynb): apply logger to notebook (2noise#410)

waywayyang · Jun 22, 2024 · 1430c95 · 1430c95
1 parent 7fecdb8
commit 1430c95
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/examples/ipynb/colab.ipynb b/examples/ipynb/colab.ipynb
@@ -50,6 +50,7 @@
         "torch.set_float32_matmul_precision('high')\n",
         "\n",
         "from ChatTTS import ChatTTS\n",
+        "from ChatTTS.tools.logger import get_logger\n",
         "from IPython.display import Audio"
       ]
     },
@@ -70,7 +71,7 @@
       },
       "outputs": [],
       "source": [
-        "chat = ChatTTS.Chat()"
+        "chat = ChatTTS.Chat(get_logger(\"ChatTTS\"))"
       ]
     },
     {

diff --git a/examples/ipynb/example.ipynb b/examples/ipynb/example.ipynb
@@ -33,6 +33,7 @@
     "torch.set_float32_matmul_precision('high')\n",
     "\n",
     "import ChatTTS\n",
+    "from tools.logger import get_logger\n",
     "from IPython.display import Audio"
    ]
   },
@@ -51,7 +52,7 @@
    "source": [
     "os.chdir(root_dir)\n",
     "\n",
-    "chat = ChatTTS.Chat()"
+    "chat = ChatTTS.Chat(get_logger(\"ChatTTS\"))"
    ]
   },
   {

diff --git a/tools/logger/log.py b/tools/logger/log.py
@@ -1,6 +1,7 @@
 import platform
 import logging
 from datetime import datetime, timezone
+from functools import lru_cache
 
 logging.getLogger("numba").setLevel(logging.WARNING)
 logging.getLogger("httpx").setLevel(logging.WARNING)
@@ -47,6 +48,7 @@ def format(self, record: logging.LogRecord):
         logstr += f"] {str(record.name)} | {str(record.msg)%record.args}"
         return logstr
 
+@lru_cache(maxsize=None)
 def get_logger(name: str, lv = logging.INFO):
     logger = logging.getLogger(name)
     syslog = logging.StreamHandler()