From 2d899ae370edd00b3d0815718e0c306ed1bb95a6 Mon Sep 17 00:00:00 2001 From: Xavier Date: Tue, 6 Sep 2022 00:13:09 -0700 Subject: [PATCH] Update README.md --- README.md | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 94ce532..4217361 100644 --- a/README.md +++ b/README.md @@ -1 +1,9 @@ -# dreambooth-stable-diffusion \ No newline at end of file +# Dreambooth on Stable Diffusion + +This is an implementtaion of Google's [Dreambooth](https://arxiv.org/abs/2208.12242) with [Stable Diffusion](https://github.com/CompVis/stable-diffusion). The original Dreambooth is based on [Imagen](https://imagen.research.google/) text-to-image model. However, neither the model nor the pre-trained weights of Imagen is available. To enable people to fine-tune a text-to-image model with a few examples, I implemented the idea of Dreambooth on Stable diffusion. + +This code repository is based on that of [Textual Inversion](https://github.com/rinongal/textual_inversion). Note that Textual Inversion only optimizes word ebedding, while dreambooth fine-tunes the whole diffusion model. + +The implementation makes minimum changes over the official codebase of Textual Inversion, and in fact some components in Textual Inversion, such as the embedding manager, are not deleted, although they will never be used here. + +##Usage