From 419b3b9846ecdeea8795a4dc3ec5b9fbaf33abbb Mon Sep 17 00:00:00 2001 From: plucked Date: Mon, 28 Aug 2023 11:14:50 +0000 Subject: [PATCH 1/2] chore: add parameters wandb ai --- library/train_util.py | 1 + 1 file changed, 1 insertion(+) diff --git a/library/train_util.py b/library/train_util.py index 3a68bed9..d76b1466 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -3081,6 +3081,7 @@ def prepare_accelerator(args: argparse.Namespace): os.environ["WANDB_DIR"] = logging_dir if args.wandb_api_key is not None: wandb.login(key=args.wandb_api_key) + wandb.init(project="Stable Diffision Training", name=args.output_name, config=args) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps, From bcb1ef057f5f00f4c02922939655253f4cfe4195 Mon Sep 17 00:00:00 2001 From: plucked Date: Mon, 28 Aug 2023 11:15:52 +0000 Subject: [PATCH 2/2] chore: add parameters wandb ai --- library/train_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/train_util.py b/library/train_util.py index d76b1466..4ae0a9a7 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -3081,7 +3081,7 @@ def prepare_accelerator(args: argparse.Namespace): os.environ["WANDB_DIR"] = logging_dir if args.wandb_api_key is not None: wandb.login(key=args.wandb_api_key) - wandb.init(project="Stable Diffision Training", name=args.output_name, config=args) + wandb.init(project="Stable Diffusion Training", name=args.output_name, config=args) accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps,