#109 · Primary category: Image Generation
Dreambooth-Stable-Diffusion
Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual Inversion (https://arxiv.org/abs/2208.01618) for Stable Diffusion (https://arxiv.org/abs/2112.10752). Tweaks focused on training faces, objects, and styles.
Project last updated:01/08/24
GitHub Stars
3.2K
Forks
526
Contributors
18
License
MIT
Why we included this project
This is a training toolkit for teaching a Stable Diffusion model to recognize and reproduce a specific face, object, or visual style. It combines the Dreambooth and Textual Inversion methods into one pipeline for fine-tuning on a small set of your own images, with tweaks aimed at keeping faces and objects recognizable rather than generic. Setup guides cover cloud GPU services and local machines, so you can get going even without prior fine-tuning experience. The author, Joe Penna, is a film director who built it to train actors, props, and locations for his movies, and that shows in the debugging notes that walk you through common failure cases like faces that look nothing like the source or training images that repeat too literally. If you want consistent characters or a signature look instead of generic outputs, this gives you the training code and the troubleshooting guidance to get there.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
stable-diffusion-webui
Stable Diffusion web UI
ComfyUI
The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
upscayl
🆙 Upscayl - #1 Free and Open Source AI Image Upscaler for Linux, MacOS and Windows.
InvokeAI
InvokeAI is a leading open-source creative engine for Stable Diffusion, offering an industry-leading web UI for generating and editing visual media.