defaults

mosaicml · Sep 13, 2024 · bec6bbb · bec6bbb
1 parent a862d6e
commit bec6bbb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -467,8 +467,8 @@ def _save_checkpoint(
         self,
         state: State,
         logger: Logger,
-        upload_to_save_folder: bool,
-        register_to_mlflow: bool,
+        upload_to_save_folder: bool = True,
+        register_to_mlflow: bool = True,
     ):
         """Save a HuggingFace formatted checkpoint.