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={