From 581643312166af2eb11d8c35fdb3da66b905c96c Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Thu, 24 Apr 2025 00:02:40 +0000 Subject: [PATCH] nit --- src/axolotl/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/axolotl/train.py b/src/axolotl/train.py index 0fe569b1e..5c73647f0 100644 --- a/src/axolotl/train.py +++ b/src/axolotl/train.py @@ -187,8 +187,6 @@ def execute_training( trainer: The configured trainer object. resume_from_checkpoint: Path to checkpoint to resume from, if applicable. """ - LOG.info("Starting trainer...") - # Define the context managers to use flash_context = ( torch.backends.cuda.sdp_kernel( @@ -209,6 +207,7 @@ def execute_training( else nullcontext() ) + LOG.info("Starting trainer...") with flash_context, sequence_parallel_context: trainer.train(resume_from_checkpoint=resume_from_checkpoint)