style

mosaicml · Dec 17, 2024 · 44e609e · 44e609e
1 parent 3a7c0f1
commit 44e609e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -840,8 +840,10 @@ def _save_checkpoint(
             assert new_model_instance is not None
             if register_to_mlflow:
                 self._register_hf_model(
-                    temp_save_dir, original_tokenizer, use_temp_dir,
-                    new_model_instance
+                    temp_save_dir,
+                    original_tokenizer,
+                    use_temp_dir,
+                    new_model_instance,
                 )
             else:
                 # Clean up the temporary directory if we don't need to register to mlflow.