add log outside of process

mosaicml · Oct 23, 2024 · 04ddfaa · 04ddfaa
1 parent 99589c7
commit 04ddfaa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -781,6 +781,7 @@ def tensor_hook(
 
                     # Spawn a new process to register the model.
                     # Slower method to register the model via log_model.
+                    log.info('USING MY BRANCH!!!!!!!!!!!!!!')
                     process = SpawnProcess(
                         target=_log_model_multiprocess,
                         kwargs={