wip

mosaicml · Mar 29, 2024 · 3906f5a · 3906f5a
1 parent a59263b
commit 3906f5a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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.