license:mit region:us
---
license: mit
---

A collection of regularization / class instance datasets for the [Stable Diffusion v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5) model to use for DreamBooth prior preservation loss training. Files labeled with "mse vae" used the [stabilityai/sd-vae-ft-mse](https://huggingface.co/stabilityai/sd-vae-ft-mse) VAE. For ease of use, datasets are stored as zip files containing 512x512 PNG images. The number of images in each zip file is specified at the end of the filename.

There is currently a bug where HuggingFace is incorrectly reporting that the datasets are pickled. They are not picked, they are simple ZIP files containing the images.

This dataset is based on the conventions setup by [Progamergov](https://huggingface.co/datasets/ProGamerGov/StableDiffusion-v1-5-Regularization-Images) and their very useful regularization images.

Currently this repository contains the following datasets (datasets are named after the prompt they used):

* "**interior design**": 2354 images generated using 50 DDIM steps and a CFG of 7, using the MSE VAE.