From dc37fd2ff6cb9ef4dbfd0b0c887aea25b20901b2 Mon Sep 17 00:00:00 2001
From: saltacc <alotofcatz@gmail.com>
Date: Sat, 22 Apr 2023 01:26:31 -0700
Subject: [PATCH] fix no logging command line arg

---
 library/train_util.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/library/train_util.py b/library/train_util.py
index d43e0075d..eb46fbb0f 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -2761,6 +2761,8 @@ def prepare_accelerator(args: argparse.Namespace):
             if logging_dir is not None:
                 os.makedirs(logging_dir, exist_ok=True)
                 os.environ["WANDB_DIR"] = logging_dir
+    else:
+        log_with = None
 
     accelerator = Accelerator(
         gradient_accumulation_steps=args.gradient_accumulation_steps,