From 7a77ee36e4e40f8c2aafa9308f818b28cd83cbf1 Mon Sep 17 00:00:00 2001
From: Milo Cress <milo.cress@databricks.com>
Date: Fri, 6 Dec 2024 00:03:42 +0000
Subject: [PATCH] hook in

---
 llmfoundry/callbacks/hf_checkpointer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py
index 945b10c823..a127553a51 100644
--- a/llmfoundry/callbacks/hf_checkpointer.py
+++ b/llmfoundry/callbacks/hf_checkpointer.py
@@ -808,6 +808,8 @@ def tensor_hook(
                                     3600,
                                 'mlflow_logging_config':
                                     self.mlflow_logging_config,
+                                'is_peft':
+                                    self.using_peft,
                             },
                         )