Adding missing audio file & by-default creating ckpts dir when not pr…

…esent (metavoiceio#100)
Pkpk11 · Mar 19, 2024 · 68f0acd · 68f0acd
1 parent 5a12716
commit 68f0acd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/data/audio.wav b/data/audio.wav
diff --git a/fam/llm/finetune.py b/fam/llm/finetune.py
@@ -40,7 +40,8 @@
 
 ckpts_base_dir = pathlib.Path(__file__).resolve().parent / "ckpts"
 if not os.path.exists(ckpts_base_dir) and master_process:
-    raise Exception(f"ckpts dir {ckpts_base_dir} does not exist!")
+    print("Checkpoints directory didn't exist, creating...")
+    ckpts_base_dir.mkdir(parents=True)
 
 if master_process:
     if "/" in out_dir: