Using DreamBooth on SD on a 3090 w/24gb VRAM (about 1.5 hrs to train)

• Dreambooth-Stable-Diffusion

  100 3,162 6.8 Jupyter Notebook

  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. (by JoePenna)

Here is a version on put on RunPod which seems to work for me so far- https://github.com/JoePenna/Dreambooth-Stable-Diffusion/

• Stable-textual-inversion_win

  15 240 10.0 Jupyter Notebook

Would it be possible for you to add this new code in the "regular" textual inversion code? like in this one : https://github.com/nicolai256/Stable-textual-inversion_win - I'm using a 3090, batch size of 3, workers 10, size 384 - works pretty good but if your modification could reduce the VRAM, it could go faster.

• WorkOS

  workos.com sponsored

  The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  

• bitsandbytes

  61 5,344 9.4 Python

  Accessible large language models via k-bit quantization for PyTorch.

https://github.com/TimDettmers/bitsandbytes https://huggingface.co/blog/hf-bitsandbytes-integration

Suggest a related project