How to Automate DreamBooth Image Generation with Runpod's API
Blog post from RunPod
DreamBooth, a tool for customizing Stable Diffusion models, enables users to train models to generate images featuring specific new concepts, such as a pet or a friend, by utilizing Runpod's DreamBooth endpoint. This process, which can be automated with API calls, requires a set of training images and a prompt, along with an API key for authorization. Users need to prepare and host a zip file of training images, ensuring the images are square and resized, typically to 512x512 pixels, and labeled with a unique keyword. By sending a POST request to Runpod's DreamBooth endpoint with the necessary parameters, including the training data URL and image generation prompt, users can initiate a job to train the model and generate images. The status of the job can be monitored through a separate GET request using the job ID, and once completed, the generated images can be accessed via URLs provided in the response. Advanced options are available, allowing users to specify different starting models or generate multiple images per request.