Update logging_config.py (#51)

epfl-dlab · Jun 4, 2024 · 9c2aaf3 · 9c2aaf3
1 parent 651dca0
commit 9c2aaf3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/transformers_cfg/logging_config.py b/transformers_cfg/logging_config.py
@@ -15,4 +15,7 @@ def setup_logging():
         "CRITICAL": logging.CRITICAL,
     }
     log_level = log_levels.get(log_level_name, logging.WARNING)
-    logging.basicConfig(level=log_level)
+    # Create a logger for the library
+    logger = logging.getLogger('transformers_cfg')
+    # the level will propagate to loggers with submodule scope
+    logger.setLevel(log_level)