mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-06 21:52:27 +00:00
fix no logging command line arg
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user