Dreambooth Gui
Provides a easy-to-use gui for users to train Dreambooth with custom images. This
Gui supports any NVIDIA card with >10GB VRAM.
![BuildStatus](https://github.com/smy20011/dreambooth-gui/actions/workflows/test.yaml/badge.svg)
Screenshots
![203261997-cc3c8456-a7fc-4162-b4fd-d953b4c5fbcd](https://user-images.githubusercontent.com/1560175/204433559-9aacdd94-32c1-445b-8fdc-6aa03a8a67a0.jpg)
Highlights
- Automatically decide training params that fit your available VRAM.
- Easy to use Gui for you to select images.
- Support prior preservation training with class images.
- Automatically cache models.
Install (Windows)
- Download and install docker from https://www.docker.com/
- Setup WSL2 for windows. https://learn.microsoft.com/en-us/windows/wsl/install
- If you find 'WSL 2 installation is incomplete' when starting docker, you can follow this video to fix it. https://www.youtube.com/watch?v=Ffzud5xLH4c
- Download and install dreambooth-gui_*_x64_en-US.msi
from release page.
- Run the dreambooth-gui as administrator.
Install (Linux)
- Download and install docker from https://www.docker.com/
- Download AppImage from release page.
- Run
chmod +x dreambooth-gui_*amd64.AppImage
- Run
sudo dreambooth-gui_*amd64.AppImage
FAQs
-
Failed to create directory
Please make sure you have the latest verion of GUI. This is a old bug that fixed in v0.1.3
-
PIL.UnidentifiedImageEnnon: cannot identify image file
Make sure the instance image folder only have image.
-
Read-only file system error
Make sure you have enough space in C(or home folder) before running the Gui.
-
Train with SD v2
Training with SD v2 is supported. However, you need to type stabilityai/stable-diffusion-2
as model name. Local v2 training is not supported right now.
-
I have other questions!
Please use the discussion page for Q&A.
I will convert FAQs to a bug if necessary. I perfer to keep the issue section clean but keep getting questions
that I answered before.
Roadmap
- [X] Refactor the state management.
- [X] Better error handling to cover FAQs.
- [ ] Allow advanced customization
- [X] Load local model.
- [ ] Save/Load config for users.
- [ ] Save model / pics in places other than $APP_DIR
- [ ] Better training progress report.
- [X] Create a dialog when training finished.
- [ ] Progress bar.
- [X] Support model converstion.
Additional Resources
- Someone in japan write a doc regarding how to use it.