From b143df3926dec89d0969faf561431b1425729fda Mon Sep 17 00:00:00 2001 From: wenfeiy-db Date: Thu, 9 Nov 2023 09:11:09 -0800 Subject: [PATCH] debug --- llmfoundry/callbacks/hf_checkpointer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py index fe50c950be..112bc07571 100644 --- 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',