Wandb integration

[cavit99]

It would be SO good to have https://wandb.ai/ integration if you're able to implement. Definetely worth ko-fi!

[hollowstrawberry]

Ah yes that would be really nice, unfortunately I'm not too familiar with this, it's unlikely I'll get into it all things considered. But I'll keep the issue here.

[cavit99]

Fair enough but I think it's actually super easy as built into kohya, just need to pass arguments
--log_with wandb
--wandb_api_key="{wandb_api_key}"

does that help?

[hollowstrawberry]

Oh I'll look into it, I was planning to update kohya soon anyway

[hollowstrawberry]

Kohya has been updated, will look into wandb soon

[Taikakim]

Yes, this would be invaluable, especially with the generation of sample images to monitor the progress of the model.

[hollowstrawberry]

I ended up adding it a while ago.