pc

mosaicml · Oct 16, 2024 · d9dbf43 · d9dbf43
1 parent 78fa9a9
commit d9dbf43
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/llmfoundry/command_utils/train.py b/llmfoundry/command_utils/train.py
@@ -592,7 +592,8 @@ def train(cfg: DictConfig) -> Trainer:
         profiler=profiler,
         compile_config=compile_config,
         spin_dataloaders=train_cfg.spin_dataloaders,
-        accumulate_train_batch_on_tokens=train_cfg.accumulate_train_batch_on_tokens,
+        accumulate_train_batch_on_tokens=train_cfg.
+        accumulate_train_batch_on_tokens,
     )
 
     _sort_callbacks(trainer)