An open API service indexing awesome lists of open source software.

https://github.com/storj/secure-stable-diffusion

Docker image for secure AI model training and inference with Stable Diffusion and Storj.
https://github.com/storj/secure-stable-diffusion

ai dreambooth secure stability-ai stable-diffusion stable-diffusion-webui storj

Last synced: 7 months ago
JSON representation

Docker image for secure AI model training and inference with Stable Diffusion and Storj.

Awesome Lists containing this project

README

          

# Secure Stable Diffusion by Storj

This Docker image focuses on secure training and inference with Stable Diffusion by securely storing and fetching datasets and checkpoints from the [Storj Decentralized Cloud Storage](https://storj.io).

## Quickstart

* RunPod:
https://runpod.io/gsc?template=3jgnr4mriw&ref=se98m3j8
* VALDI: [Start and connect to a Direct VM](https://github.com/storj/secure-stable-diffusion/wiki/VALDI%3A-Start-and-connect-to-a-Direct-VM)

## Included in this Docker image

* [Stable Diffusion Web UI](https://github.com/AUTOMATIC1111/stable-diffusion-webui) 1.5.1
* [Stable Studio](https://github.com/Stability-AI/StableStudio)
* Automated fetching of AI models from your Storj account on startup
* Jupyter Notebook for [DreamBooth](https://dreambooth.github.io/) training and storing the trained checkpoints to your Storj account.

## Environment variables

| Var name | Description |
| -------- | ----------- |
| STORJ_ACCESS | The [access grant](https://docs.storj.io/dcs/concepts/access/access-grants) to use for transferring files from and to your Storj DCS account |
| STORJ_BUCKET | The bucket to use for transferring files from and to your Storj DCS account |
| STORJ_MODEL_PATH | The object key to a specific AI model or a prefix/folder with multiple AI models to automatically download at startup |

If the Docker container is configured with all three environment variables, it will automatically download the AI models at startup.

## How to use

The usage instructions are maintained in the [wiki](https://github.com/storj/secure-stable-diffusion/wiki).

### Instructions for [RunPod](https://runpod.io/?ref=se98m3j8)
* [Deploy and launch a pod](https://github.com/storj/secure-stable-diffusion/wiki/RunPod%3A-Deploy-and-launch-a-pod)
* [Configure the pod to download your AI models from Storj automatically](https://github.com/storj/secure-stable-diffusion/wiki/RunPod%3A-Configure-the-pod-to-download-your-AI-models-from-Storj-automatically)
* [Launch the Stable Diffusion Web UI](https://github.com/storj/secure-stable-diffusion/wiki/RunPod%3A-Launch-the-Stable-Diffusion-Web-UI)
* [Launch the Stable Studio](https://github.com/storj/secure-stable-diffusion/wiki/RunPod%3A-Launch-the-Stable-Studio)
* [Train your model and upload it to Storj](https://github.com/storj/secure-stable-diffusion/wiki/RunPod%3A-Train-your-model-and-upload-it-to-Storj)

### Instructions for [VALDI](https://valdi.ai/signup?ref=LdwHNPBZ)
* [Start and connect to a Direct VM](https://github.com/storj/secure-stable-diffusion/wiki/VALDI%3A-Start-and-connect-to-a-Direct-VM)
* [Launch the Docker container in inference mode](https://github.com/storj/secure-stable-diffusion/wiki/VALDI%3A-Launch-the-Docker-container-in-inference-mode)
* [Launch the Stable Diffusion Web UI](https://github.com/storj/secure-stable-diffusion/wiki/VALDI%3A-Launch-the-Stable-Diffusion-Web-UI)
* [Launch the Stable Studio](https://github.com/storj/secure-stable-diffusion/wiki/VALDI%3A-Launch-the-Stable-Studio)
* [Train your model and upload it to Storj](https://github.com/storj/secure-stable-diffusion/wiki/VALDI%3A-Train-your-model-and-upload-it-to-Storj)