lint

mosaicml · Jan 11, 2024 · 6def6ca · 6def6ca
1 parent b8156a6
commit 6def6ca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -438,8 +438,10 @@ def main(cfg: DictConfig) -> Trainer:
             format=
             f'%(asctime)s: rank{dist.get_global_rank()}[%(process)d][%(threadName)s]: %(levelname)s: %(name)s: %(message)s'
         )
-        logging.getLogger('llmfoundry').setLevel(python_log_level.upper())  # Foundry module
-        logging.getLogger(__name__).setLevel(python_log_level.upper())  # Train script
+        logging.getLogger('llmfoundry').setLevel(
+            python_log_level.upper())  # Foundry module
+        logging.getLogger(__name__).setLevel(
+            python_log_level.upper())  # Train script
 
     # Initialize context
     init_context = process_init_device(model_config, fsdp_config)