fix no logging command line arg

DarksealStudios · Apr 22, 2023 · dc37fd2 · dc37fd2
1 parent f256660
commit dc37fd2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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,