Home / Companies / RunPod / Blog / Post Details
Content Deep Dive

How to Automate DreamBooth Image Generation with Runpod's API

Blog post from RunPod

Post Details
Company
Date Published
Author
Pardeep Singh
Word Count
1,289
Language
English
Hacker News Points
-
Summary

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.