diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
index 006c245213..13217f301e 100644
--- a/llmfoundry/callbacks/hf_checkpointer.py
+++ b/llmfoundry/callbacks/hf_checkpointer.py
@@ -203,6 +203,7 @@ def _register_mlflow_model(
             model_saving_kwargs['transformers_model'] = components
             model_saving_kwargs.update(logging_config)
 
+        print("HEY", model_saving_kwargs)
         mlflow_logger.save_model(**model_saving_kwargs)
 
         # Upload the license file generated by mlflow during the model saving.