Dreambooth able to run on 18GB VRAM now. Potentially on 16GB in the future with more optimization.

This page summarizes the projects mentioned and recommended in the original post on reddit.com/r/StableDiffusion

Our great sponsors
  • Scout APM - Truly a developer’s best friend
  • InfluxDB - Build time-series-based applications quickly and at scale.
  • Sonar - Write Clean Python Code. Always.
  • Zigi - Workflow assistant built for devs & their teams
  • diffusers

    🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch (by ShivamShrirao)

    Repo located here: https://github.com/ShivamShrirao/diffusers/tree/main/examples/dreambooth

  • xformers

    Hackable and optimized Transformers building blocks, supporting a composable construction.

    Stable Diffusion dreambooth training in just 17.7GB GPU usage by replacing the attention with memory efficient flash attention from xformers. Along with using way less memory, it also runs 2 times faster. So it's possible to train SD in 24GB GPUs now.

  • Scout APM

    Truly a developer’s best friend. Scout APM is great for developers who want to find and fix performance issues in their applications. With Scout, we'll take care of the bugs so you can focus on building great things 🚀.

  • nvtop

    GPUs process monitoring for AMD, Intel and NVIDIA

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

    I tried the fork of JoePenna, there is a notebook with all required step but it's tailored for another GPU cloud rental architecture. I didn't manage to make it work but you can give it a try

  • Pytorch

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

  • InfluxDB

    Build time-series-based applications quickly and at scale.. InfluxDB is the Time Series Data Platform where developers build real-time applications for analytics, IoT and cloud-native services in less time with less code.

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts