From 04ddfaac43aa125d7dadf98b05b869c8bdf6398e Mon Sep 17 00:00:00 2001
From: Nancy Hung <nancy.hung1068@gmail.com>
Date: Wed, 23 Oct 2024 11:41:00 -0700
Subject: [PATCH] add log outside of process

---
 llmfoundry/callbacks/hf_checkpointer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
index 8380f7917c..fe0847c3b5 100644
--- 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={