This guide provides you steps to deploy a Stable Diffusion WebUI solution in your Google Cloud Project.
| Folder | Description |
|---|---|
| Stable-Diffusion-UI-Agones | Demo with all the YAML files and Dockerfiles for hosting Stable Diffusion WebUI using Agones. |
| Stable-Diffusion-UI-GKE | Demo with all the YAML files and Dockerfiles for hosting Stable Diffusion WebUI using GKE. |
| Stable-Diffusion-Vertex | Reference codes for DreamBooth & Lora training on Vertex AI |
| terraform-provision-infra | Terraform scripts and resources to create the demo environment. |
| examples | Example folder for a working directory |
This project demos how to effectively host the popular AUTOMATIC1111 web interface Stable-Diffusion-WebUI. This is for demo purpose, you may need minimum modification according to your needs before put into production. However, it could also be use directly as an internal project.
Projects and products include:
- GKE for hosting Stable Diffusion and attaching GPU hardware to nodes in your Kubernetes cluster.
- Filestore for saving models and output files.
- Vertex AI for training and fine-tuning the model.
- Cloud Build for building images and Continuous Integration.
- GKE Standard clusters running Agones for isolating runtime for different users and scaling.
- Stable Diffusion for generating images from text.
- Webui: A browser interface for Stable Diffusion.
- Recommended for most use cases, use dedicated pod+gpu, with (almost) the same experience as running on your own workstation.
- Architecture GKE + Agones + Spot(optional) + GPU(optional time sharing) + Vertex AI for supplementary Dreambooth/Lora training
- Use Cloud identity-aware proxy for login and authentication with Google account
- A demo nginx+lua implementation as a frontend UI to interactive with Agones
- Using Agones for resource allocation and release instead of HPA
- Run Inference, training and all other functions and extensions on WebUI
- Supplementary Dreambooth/Lora Training on Vertex AI
- No intrusive change against AUTOMATIC1111 webui, easy to upgrade or install extensions with Dockerfile
- Recommended for serving as a Saas platform
- Architecture GKE + GPU(optional time sharing) + Spot(optional) + HPA + Vertex AI for supplementary Dreambooth/Lora training
- No conflicts for multiple users, one deployment per model, use different mount point to distinguish models
- Scaling with HPA with GPU metrics
- Inference on WebUI, but suitable for training
- Supplementary Dreambooth/Lora Training on Vertex AI
- No intrusive change against AUTOMATIC1111 webui, easy to upgrade or install extensions with Dockerfile
For Stable-Diffusion-UI-Agones, it support multi users/sessions in nature since it assign a dedicated pod for each login user. For Stable-Diffusion-UI-GKE, AUTOMATIC1111's Stable Diffusion WebUI does not support multi users/sessions at this moment, you can refer to AUTOMATIC1111/stable-diffusion-webui#7970. To support multi-users, we create one deployment for each model.
For Stable-Diffusion-UI-Agones, in the demo we use init script to initialize folders for each users. You can customize the init script to meet your need, and there is a reference.
For Stable-Diffusion-UI-GKE, instead of building images for each model, we use one image with shared storage from Filestore and properly orchestrate for our files and folders. Please refer to the deployment_*.yaml for reference.
Your folder structure could probably look like this in your Filestore file share, you may have to adjust according to your needs:
/models/Stable-diffusion # <--- This is where Stable Diffusion WebUI looking for models
|-- nai
| |-- nai.ckpt
| |-- nai.vae.pt
| `-- nai.yaml
|-- sd15
| `-- v1-5-pruned-emaonly.safetensors
/inputs/ # <--- for training images, only use it when running training job from UI(sd_dreammbooth_extension)
|-- alvan-nee-cropped
| |-- alvan-nee-9M0tSjb-cpA-unsplash_cropped.jpeg
| |-- alvan-nee-Id1DBHv4fbg-unsplash_cropped.jpeg
| |-- alvan-nee-bQaAJCbNq3g-unsplash_cropped.jpeg
| |-- alvan-nee-brFsZ7qszSY-unsplash_cropped.jpeg
| `-- alvan-nee-eoqnr8ikwFE-unsplash_cropped.jpeg
/outputs/ # <--- for generated images
|-- img2img-grids
| `-- 2023-03-14
| |-- grid-0000.png
| `-- grid-0001.png
|-- img2img-images
| `-- 2023-03-14
| |-- 00000-425382929.png
| |-- 00001-631481262.png
| |-- 00002-1301840995.png
We made an example script to work as an extension for file upload. Besides, you can use extensions for image browsing and downloading(https://github.com/zanllp/sd-webui-infinite-image-browsing), model/lora downloading(https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper) and more.
HPA & Agones only allow at least one replica, to do this you will have to manually scale to 0 or delete the resource. e.g. For GKE,
kubectl scale --replicas=1 deployment stable-diffusion-train-deployment
for Agones,
kubectl delete fleet sd-agones-fleet