diff --git a/README.md b/README.md index 5ce7bfb..b8f42c6 100644 --- a/README.md +++ b/README.md @@ -2,10 +2,10 @@ Run Stable Diffusion with companion models on a GPU-enabled Kubernetes Cluster - complete with a [WebUI](https://github.com/hlky/stable-diffusion-webui) and automatic model fetching for a 2 step install that takes less than 2 minutes (excluding download times). -If you are on a restricted internet connection and want to save bandwidth, you can also [manually build the dockerfile](./docker/stable-diffusion-serving.Dockerfile) and push it to a local container registry. - Uses the `nvidia/cuda` image as a base. +![Screenshot of the Stable Diffusion UI](img/screenshot.png) + ### Features - Automatic Model Fetching @@ -23,11 +23,12 @@ Uses the `nvidia/cuda` image as a base. ## Setup - Add the helm repo with `helm repo add amithkk-sd https://amithkk.github.io/stable-diffusion-k8s` +- Fetch latest charts with `helm repo update` - (Optional) Create your own [`values.yaml`](./charts/stable-diffusion/values.yaml) with customized settings - Some things that you might want to change could include the `nodeAffinity`, `cliArgs` (see below) and `ingress` settings (that will allow you to access this externally without needing to `kubectl port-forward`) -- Install with `helm install amithkk-sd/stable-diffusion -f ` +- Install with `helm install --generate-name amithkk-sd/stable-diffusion -f ` -Wait for the containers to come up and follow the instructions returned by Helm to connect. This may take a while as it has to download a ~10GiB docker image and ~5Gib of models +Wait for the containers to come up and follow the instructions returned by Helm to connect. This may take a while as it has to download a ~5GiB docker image and ~5GiB of models ## Config diff --git a/charts/stable-diffusion/Chart.yaml b/charts/stable-diffusion/Chart.yaml index 86dd3a9..2902031 100644 --- a/charts/stable-diffusion/Chart.yaml +++ b/charts/stable-diffusion/Chart.yaml @@ -15,7 +15,7 @@ type: application # This is the chart version. This version number should be incremented each time you make changes # to the chart and its templates, including the app version. # Versions are expected to follow Semantic Versioning (https://semver.org/) -version: "1.0.4" +version: "1.0.5" # This is the version number of the application being deployed. This version number should be # incremented each time you make changes to the application. Versions are not expected to diff --git a/charts/stable-diffusion/templates/statefulset.yaml b/charts/stable-diffusion/templates/statefulset.yaml index 8e78411..195dd7a 100644 --- a/charts/stable-diffusion/templates/statefulset.yaml +++ b/charts/stable-diffusion/templates/statefulset.yaml @@ -68,14 +68,7 @@ spec: volumeMounts: - mountPath: /models name: {{ include "stable-diffusion.fullname" . }}-model-store - livenessProbe: - httpGet: - path: / - port: http - readinessProbe: - httpGet: - path: / - port: http + # Todo - Implement an efficient readiness and liveness check resources: {{- toYaml .Values.resources | nindent 12 }} {{- with .Values.nodeSelector }} diff --git a/img/screenshot.png b/img/screenshot.png new file mode 100644 index 0000000..06a214c Binary files /dev/null and b/img/screenshot.png differ