I used the code from your dev branch for wandb training and adapted it to add the ability to track your training in Weights and Biases. I added "Kohya-ss" as a prefix to the project name so it'll be easier to find but feel free to change that if you'd like.
I used the code from your dev branch for wandb training and adapted it to add the ability to track your training in Weights and Biases. I added "Kohya-ss" as a prefix to the project name so it'll be easier to find but feel free to change that if you'd like.