Add some more debug logs

mosaicml · Aug 14, 2024 · a177b1e · a177b1e
1 parent 9fd1844
commit a177b1e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -125,13 +125,17 @@ def _register_model_with_run_id_multiprocess(
         )
         logging.getLogger('composer').setLevel(composer_logging_level)
 
+    log.debug('Starting registering model.')
+
     # Register model.
     mlflow_logger.register_model_with_run_id(
         model_uri=model_uri,
         name=name,
         await_creation_for=await_creation_for,
     )
 
+    log.debug('Finished registering model.')
+
 
 class HuggingFaceCheckpointer(Callback):
     """Save a huggingface formatted checkpoint during training.