only set level log for zero band

PrimeIntellect-ai · Sep 22, 2024 · 7846ee4 · 7846ee4
1 parent 8553a73
commit 7846ee4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/zeroband/utils/logging.py b/src/zeroband/utils/logging.py
@@ -28,9 +28,9 @@ def get_logger():
 
     if world_info.local_rank == 0:
         log_level = os.getenv("ZERO_BAND_LOG_LEVEL", "INFO")
-        logging.basicConfig(level=getattr(logging, log_level, logging.INFO))
+        logger.setLevel(level=getattr(logging, log_level, logging.INFO))
     else:
-        logging.basicConfig(level=logging.CRITICAL)  # Disable logging for non-zero ranks
+        logger.setLevel(level=logging.CRITICAL)  # Disable logging for non-zero ranks
 
     handler = logging.StreamHandler()
     handler.setFormatter(CustomFormatter(world_info.local_rank))