debug

mosaicml · Nov 9, 2023 · b143df3 · b143df3
1 parent d97e4af
commit b143df3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
@@ -266,7 +266,7 @@ def _save_checkpoint(self, state: State, logger: Logger):
                         # TODO: Remove after mlflow fixes the bug that makes this necessary
                         import mlflow
                         mlflow.store._unity_catalog.registry.rest_store.get_feature_dependencies = lambda *args, **kwargs: ''
-                        print(f"_save_checkpoint::self.mlflow_logging_config={self.mlflow_logger}\n[{type(self.mlflow_logger)}]")
+                        print(f"_save_checkpoint::self.mlflow_logging_config={self.mlflow_logging_config}\n[{type(self.mlflow_logging_config)}]")
                         print(f"_save_checkpoint::self.mlflow_logging_config['metadata']={self.mlflow_logging_config['metadata']}\n{type(self.mlflow_logging_config['metadata'])}]")
                         mlflow_logger.save_model(
                             flavor='transformers',