From 4301ceb15dd72d205d3889ea6770903bb4076c59 Mon Sep 17 00:00:00 2001 From: Milo Cress Date: Tue, 17 Dec 2024 14:38:10 -0500 Subject: [PATCH] style III --- llmfoundry/callbacks/hf_checkpointer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py index 5eb6d8ca98..6fa35c4980 100644 --- a/llmfoundry/callbacks/hf_checkpointer.py +++ b/llmfoundry/callbacks/hf_checkpointer.py @@ -619,8 +619,7 @@ def tensor_hook( hooks = [] for _, module in state_dict_model.named_modules(): - hooks.append( - module._register_state_dict_hook(tensor_hook),) + hooks.append(module._register_state_dict_hook(tensor_hook),) state_dict = get_model_state_dict( state_dict_model,