hook in

mosaicml · Dec 6, 2024 · 7a77ee3 · 7a77ee3
1 parent c4af194
commit 7a77ee3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,
                             },
                         )