AI-Horde icon indicating copy to clipboard operation
AI-Horde copied to clipboard

Using stable horde to generate dreambooth models

Open Tophness opened this issue 3 years ago • 2 comments

Google Dreambooth has come to SD but it currently requires 24gb of vram, making it only accessible to people who pay for a server. As SD itself runs on even 2gb cards now, wouldn't this be the perfect additional use for a server farm like this?

Tophness avatar Sep 30 '22 23:09 Tophness

Sure. The horde can support multiple models (check the KoboldAI horde). But I don't have the necessary AI understanding (or time) to implement the worker code for this myself. If there is an existing worker implementation in python however, I can modify it with bridge code and allow it to join the horde.

As always, PRs are welcome

db0 avatar Sep 30 '22 23:09 db0

Not sure if the implementation/etc would be compatible, but here's another distributed StableDiffusion training project a friend recently linked me to:

  • https://github.com/chavinlo/distributed-diffusion
    • Which seems to build on top of: https://github.com/learning-at-home/hivemind
      • Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

There's also some examples of dreambooth training from the diffusers repo:

  • https://github.com/huggingface/diffusers/tree/main/examples
    • https://github.com/huggingface/diffusers/tree/main/examples/dreambooth

0xdevalias avatar Nov 06 '22 01:11 0xdevalias

This issue dates back to a different era of the horde. Closing as stale.

tazlin avatar Aug 26 '24 18:08 tazlin