diff --git a/library/train_util.py b/library/train_util.py index d43e0075..eb46fbb0 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,