Dreambooth Stable Diffusion training in just 12.5 GB VRAM, using the 8bit adam optimizer from bitsandbytes along with xformers while being 2 times faster.

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

Our great sponsors
  • WorkOS - The modern identity platform for B2B SaaS
  • InfluxDB - Power Real-Time Data Analytics at Scale
  • SaaSHub - Software Alternatives and Reviews
  • diffusers

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

  • had to add !pip install git+https://github.com/ShivamShrirao/diffusers.git to get the correct diffusers version to start training

  • I also made a wheels version now that I think should work on P100 too here: https://github.com/metrolobo/xformers_wheels/releases/download/1d31a3ac_various_6/xformers-0.0.14.dev0-cp37-cp37m-linux_x86_64.whl but not tested on any P100 as I just have the free one.

  • WorkOS

    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.

    WorkOS logo
  • diffusers

    🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.

  • theres ckpt to diffusers. I dont have enough knowledge to reverse the script so its diffusers to ckpt https://github.com/huggingface/diffusers/tree/main/scripts

  • 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)

  • Dreambooth-Stable-Diffusion-cpu

    Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion

  • Just found this one: https://github.com/andreae293/Dreambooth-Stable-Diffusion-cpu

  • huggingface_hub

    The official Python client for the Huggingface Hub.

  • OSError: We couldn't connect to 'https://huggingface.co' to load this model, couldn't find it in the cached files and it looks like /content/models/imv is not the path to a directory containing a model_index.json file. Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/diffusers/installation#offline-mode'.

  • xformers

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

  • Cloning https://github.com/facebookresearch/xformers (to revision 1d31a3a) to /tmp/pip-install-_ed1j31_/xformers_536ced8f9a164a80bc61b1bdc77f3106

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

    InfluxDB logo
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