Dreambooth-WebUI

KO | EN

gitlite — search

#nodejs #javascript #java #python #docker #jquery #raspberry-pi #express #monitoring #golang #github #mean-stack

Dreambooth-WebUI

★ 33 Open GitHub ↗

Dreambooth Stable Diffusion Demo w/ a GUI ::: Based on JoePenna's Implementation

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Dreambooth Stable Diffusion Demo w/ a GUI ::: Based on [JoePenna's Implementation](https://github.com/JoePenna/Dreambooth-Stable-Diffusion) 🤗
### Setup/Configure/Regularization/Fine-Tune: with this DreamBooth stable diffusion demo.

https://user-images.githubusercontent.com/48079849/199377219-7ff99023-1620-4182-bf94-8c0846b65de1.mp4

## BEFORE continuing. Please Verify the following pre-requisites:
    - the system is running a NVIDIA gpu
    - the gpu has ~24 GBs for fine-tuning
    - python is installed
    - NVIDIA drivers are up to date
    - the system has a compatible version of the CUDA toolkit installed
    - the system has a compatible version of the cuDNN toolkit installed

Please make sure to install python: [Python Official Website](https://www.python.org/downloads/)
1. And add it to PATH
2. Install the packages with:
    -     ./setup.sh
3. Run the UI:
    -     python webui.py

- GUI Features a Model & Dataset Configuration tab, and Model Training tab

### Additional planned features:
- [X] The Pruning Feature from the repo (12GB to 2GB)
- [X] Resolve Gradio Components not updating Components for real-time logging feed
- [ ] Add realtime graphs for tracking loss & other metrics
- [ ] Separate Mode to run the Diffusers Dreambooth with UI (resolves *most* gpu memory constraints) <---> *(In Next Update)*
    - [ ] Option 1: [fast-stable-diffusion](https://github.com/TheLastBen/fast-stable-diffusion)
    - [ ] Option 2: [ShivamShrirao-diffusers](https://github.com/ShivamShrirao/diffusers/tree/main/examples/dreambooth)
- [ ] Insert support for jupyter notebook & google colab
    - [X] Notebook finished & tested successfully
    - [ ] Adding [https://github.com/localtunnel/localtunnel](https://github.com/localtunnel/localtunnel) to view the GUI from browser if using a cloud computing instance 
- [X] Add model resume training from checkpoint
- [X] Merge multiple datasets into one
- [X] Custom Preset Configurations to Add/Delete/Load
- [X] Custom Training Job Scheduler
- [ ] Progress Bar for image generation & training
- [ ] An Advanced Image Cropping Tool using YOLOv7
- [ ] Multi-GPU support:
    - [ ] Model Parallelism
    - [ ] Data Parallelism
- [ ] Utilizing Other Sampling Algorithms
- [ ] Windows Support

Dreambooth-WebUI

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README