diff --git a/shared_utils/common_utils.py b/shared_utils/common_utils.py index fcc5727..4d42900 100644 --- a/shared_utils/common_utils.py +++ b/shared_utils/common_utils.py @@ -104,7 +104,7 @@ def parse_save_filename(save_path, output_directory, supported_extensions, class # replace time date format to current time now = datetime.now() # current date and time - all_date_format = ["%Y", "%m", "%d", "%M", "%S", "%f"] + all_date_format = ["%Y", "%m", "%d", "%H", "%M", "%S", "%f"] for date_format in all_date_format: if date_format in filename: filename = filename.replace(date_format, now.strftime(date_format)) @@ -134,4 +134,4 @@ def resume_or_download_model_from_hf(checkpoints_dir_abs, default_repo_id, model from huggingface_hub import hf_hub_download hf_hub_download(repo_id=default_repo_id, local_dir=checkpoints_dir_abs, filename=model_name, repo_type="model") - return ckpt_path \ No newline at end of file + return ckpt_path