From a061ac98018042053ff20cc445034c2f2e612ad4 Mon Sep 17 00:00:00 2001
From: Derek Paschal <dtpaschal@gmail.com>
Date: Thu, 2 Feb 2023 16:36:03 -0600
Subject: [PATCH] Adding docker capabilites to repo

---
 Dockerfile         | 10 ++++++++++
 docker-compose.yml | 23 +++++++++++++++++++++++
 2 files changed, 33 insertions(+)
 create mode 100644 Dockerfile
 create mode 100644 docker-compose.yml

diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 00000000..ef1e7f77
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,10 @@
+FROM python:3.10
+
+WORKDIR /sd
+
+EXPOSE 7860
+
+RUN apt update \
+    && apt install -y wget git htop
+
+CMD [ "./webui.sh", "-f" ]
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 00000000..869530c4
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,23 @@
+version: '3.4'
+services:
+  sd-docker:
+    container_name: sd-docker
+    image: git.exile.place/derek/sd-docker:latest
+    restart: "no"
+    ports:
+      - "7860:7860"
+    volumes:
+      - ./:/sd
+    deploy:
+      resources:
+        reservations:
+          devices:
+              - driver: nvidia
+                device_ids: ['0']
+                capabilities: [gpu]
+    healthcheck:
+      test: "wget --no-verbose --tries=1 --spider http://sd-docker:7860 || exit 1"
+      interval: 2m
+      timeout: 30s
+      retries: 5
+      start_period: 5m
\ No newline at end of file