Guides: Ready-to-go ML Environments

Ready-to-go DreamBooth Environment

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!


  1. Sign up for an account

  2. It'll redirect you to an environment creation page pre-configured with the defaults you need. (We recommend sticking with the default GPU).

  3. 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 🙂

  4. Hit create!

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:

huggingface-cli login

It'll prompt you to add your huggingface token (make sure you've accepted the Hugging Face license agreement).

Then run the training job:

sh launch.sh

(this should take about 5 minutes)

Generating samples:

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:

imgimg

If you thought that was easy, check out our other docs. We've been building the easiest way to do develop on the cloud.

Previous
Stable Diffusion