From 3b9d780043b32b6823296f2c026411d525170320 Mon Sep 17 00:00:00 2001 From: Abhishek Date: Wed, 25 Sep 2024 19:15:04 -0400 Subject: [PATCH 1/2] fix: output_dir doesn't exist during resume_from_checkpoint Signed-off-by: Abhishek --- build/accelerate_launch.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/build/accelerate_launch.py b/build/accelerate_launch.py index 0a0cfa75..eb7892ea 100644 --- a/build/accelerate_launch.py +++ b/build/accelerate_launch.py @@ -98,6 +98,8 @@ def main(): # ########## output_dir = job_config.get("output_dir") + if not os.path.exists(output_dir): + os.makedirs(output_dir) try: # checkpoints outputted to tempdir, only final checkpoint copied to output dir launch_command(args) From aed6de2f2f019ac386243662c555bf8ced216b8d Mon Sep 17 00:00:00 2001 From: Abhishek Date: Wed, 25 Sep 2024 19:17:52 -0400 Subject: [PATCH 2/2] fix: fmt Signed-off-by: Abhishek --- build/accelerate_launch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/build/accelerate_launch.py b/build/accelerate_launch.py index eb7892ea..6cbc7d25 100644 --- a/build/accelerate_launch.py +++ b/build/accelerate_launch.py @@ -98,7 +98,7 @@ def main(): # ########## output_dir = job_config.get("output_dir") - if not os.path.exists(output_dir): + if not os.path.exists(output_dir): os.makedirs(output_dir) try: # checkpoints outputted to tempdir, only final checkpoint copied to output dir