Merge bcb1ef057f into 52c8dec953

2026-04-15 08:36:41 +00:00 · 2025-04-08 00:36:34 +08:00
parent 52c8dec953 bcb1ef057f
commit cb41f76281
1 changed files with 1 additions and 0 deletions
--- 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"