From d25b9661af92b035021de510ae1bb8b098ae760c Mon Sep 17 00:00:00 2001
From: Victor Hall <victor.charles.hall@gmail.com>
Date: Sat, 15 Apr 2023 13:27:12 -0400
Subject: [PATCH] fix bug in optimizer path

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

diff --git a/train.py b/train.py
index e1d90f4..b9a1aba 100644
--- a/train.py
+++ b/train.py
@@ -438,8 +438,8 @@ def main(args):
             logging.info(f" Saving optimizer state to {save_path}")
             save_optimizer(optimizer, optimizer_path)
 
+    optimizer_state_path = None
     try:
-
         # check for a local file
         hf_cache_path = get_hf_ckpt_cache_path(args.resume_ckpt)
         if os.path.exists(hf_cache_path) or os.path.exists(args.resume_ckpt):