diff --git a/llmfoundry/command_utils/train.py b/llmfoundry/command_utils/train.py index 2168cfefd4..f647565386 100644 --- a/llmfoundry/command_utils/train.py +++ b/llmfoundry/command_utils/train.py @@ -234,6 +234,9 @@ def train(cfg: DictConfig) -> Trainer: logging.getLogger(__name__).setLevel( train_cfg.python_log_level.upper(), ) # Train script + logging.getLogger('streaming').setLevel( + train_cfg.python_log_level.upper(), + ) # Streaming module _initialize_dist_with_barrier(dist_timeout=train_cfg.dist_timeout)