From 9efdf56e38cc87938bfabce4eb1b28f51f4aec3b Mon Sep 17 00:00:00 2001
From: Zach Nussbaum <zanussbaum@gmail.com>
Date: Sat, 8 Apr 2023 20:56:13 +0000
Subject: [PATCH] fix: saving name

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 72e53f4d..189df41c 100644
--- a/train.py
+++ b/train.py
@@ -192,7 +192,7 @@ def train(accelerator, config):
             accelerator.print(f"Failed to push to hub")
 
         unwrapped_model.save_pretrained(
-            f"{config['output_dir']}/-epoch_{epoch}",
+            f"{config['output_dir']}/epoch_{epoch}",
             is_main_process=accelerator.is_main_process,
             save_function=accelerator.save,
             state_dict=accelerator.get_state_dict(model),