In this guide, we show you how easy it is to launch and run Dreambooth on GPUs with Brev. We've pre-configured a Brev environment for Dreambooth so you won't have any hassle to get setup!
DreamBooth is a way to fine-tune the Stable Diffusion model so that you can generate samples of anything you want (such as yourself)!
If you'd rather not read, we've also made a Youtube Video for this!
To get started, hit this link to create a new Brev environment
Sign up for an account
It'll redirect you to an environment creation page pre-configured with the defaults you need. (We recommend sticking with the default GPU).
At the bottom, add a payment method...we give you 30 minutes free but we just need to make sure people don't abuse our systems 🙂
Open your new Dreambooth Brev environment:
brev open dreambooth --wait
If you don't have the Brev CLI, you can install it here
Running the training job:
Login to HuggingFace:
It'll prompt you to add your huggingface token (make sure you've accepted the Hugging Face license agreement).
Then run the training job:
(this should take about 5 minutes)
We're ready to go! Run this with whichever prompt you want:
conda activate diffusers python inference.py "fine-tuned-model-output/800" "a photo of sks dog wearing sunglasses"
And boom! Give the model a couple of minutes to load and you should see what your model generated in output1.png, output2.png, output3.png and output4.png.
Here are some we generated of our CEO:
If you thought that was easy, check out our other docs. We've been building the easiest way to do develop on the cloud.