wandb.init location in kohya_ss code??

[rafstahelin]

I’m using wandb to log Kohya training runs, and I want to add some metadata in the wandb.init so I can sort and group my logs. I I can’t find wandb.init anywhere, only wandb.login here in the train_utils.py. Where is wandb.init?