diff --git a/library/train_util.py b/library/train_util.py index 100ef475..78da917f 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -4614,6 +4614,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 Diffusion Training", name=args.output_name, config=args) # torch.compile のオプション。 NO の場合は torch.compile は使わない dynamo_backend = "NO"