From f16e9da3453c2d4e989f89e5f536262ab1148bfe Mon Sep 17 00:00:00 2001 From: Anna Pfohl Date: Wed, 10 Jan 2024 19:49:26 +0000 Subject: [PATCH] fix format --- llmfoundry/callbacks/hf_checkpointer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llmfoundry/callbacks/hf_checkpointer.py b/llmfoundry/callbacks/hf_checkpointer.py index 06fb541aec..8b139c0c25 100644 --- a/llmfoundry/callbacks/hf_checkpointer.py +++ b/llmfoundry/callbacks/hf_checkpointer.py @@ -96,7 +96,8 @@ def __init__( self.huggingface_folder_name_fstr = os.path.join( 'huggingface', huggingface_folder_name) - self.save_interval: Time = Time.from_input(save_interval, TimeUnit.EPOCH) + self.save_interval: Time = Time.from_input(save_interval, + TimeUnit.EPOCH) self.check_interval = create_interval_scheduler( self.save_interval, include_end_of_training=True) self.remote_ud = maybe_create_remote_uploader_downloader_from_uri(