VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-15 00:32:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: add parameters wandb ai

Browse Source
This commit is contained in:
plucked
2023-08-28 11:15:52 +00:00
parent 419b3b9846
commit bcb1ef057f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -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,