1 files changed, 20 insertions, 0 deletions
diff --git a/compose.yaml b/compose.yaml
new file mode 100644
index 0000000..c0f7ba1
--- /dev/null
+++ b/compose.yaml
@@ -0,0 +1,20 @@
+services:
+  imggen:
+    build: .
+    image: imggen:local
+    user: "${IMGGEN_UID:-1000}:${IMGGEN_GID:-100}"
+    group_add:
+      - video          # gid for /dev/dri/card* access as non-root
+    devices:
+      - /dev/dri
+    volumes:
+      - ${IMGGEN_OUT:-./out}:/out
+      - ${IMGGEN_CACHE:-./models}:/models
+    environment:
+      - HF_HOME=/models
+      - IMGGEN_MODEL=${IMGGEN_MODEL:-sdxl}
+      - IMGGEN_PORT=${IMGGEN_PORT:-8765}
+    ports:
+      - "127.0.0.1:${IMGGEN_PORT:-8765}:8765"
+    entrypoint: ["python3"]
+    command: ["server.py"]