Weights & Biases

Monitor your fine-tuning run with Weights & Biases
We recommend using Weights & Biases to monitor your fine-tuning runs, especially for text-to-image models like Stable Diffusion. This enables you to view the progress of your model and make sure that interim results during fine-tuning are trending in the direction you want.
To enable the integration, you'll need to add two parameters to your FineTuningConfig:
* Pass `wandb_api_key="YOUR_API_KEY"` to enable the integration.
* Use `image_log_steps` to control how many steps you see images for
Then, log in to your Weights & Biases account to see detailed fine-tuning logs.
Last modified 3mo ago