Stable Diffusion Dreambooth model template
This is the template used by replicate.com dreambooth api to build custom dreambooth models.
This template is based on cog-stable-diffusion which uses diffusers.
Usage
This template is primarily intended for use by Replicate's DreamBooth API, which is what you probably want to use to train and publish your own model.
If you really want to use this template locally, you can do so by following these steps:
- Generate some weights, put them in
weights/(use our trainer or your own) - Download NSFW safety_checker weights using
script/download-weights - Install cog & docker
- Build
cog build - Predict
cog predict -i prompt="photo of zzz" -i seed=42...