From a436ca7f053d87be76cf7fff26817b381dabce07 Mon Sep 17 00:00:00 2001
From: DHBaek <baek@comcom.ai>
Date: Fri, 20 Dec 2024 16:34:54 +0900
Subject: [PATCH] feat: add docker compose

---
 docker-compose.yml | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 docker-compose.yml

diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..1e4d1ba
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,27 @@
+version: '3'
+
+services:
+  swinlr-large-x4:
+    image: devainetwork/sr-worker
+    container_name: swinlr-large-x4-worker
+    env_file:
+      - .env
+    environment:
+      - VHOST_NAME=swinlr-large-x4
+    volumes:
+      - <firebase_credential_path>:/app/key
+      - <model_local_path>:/app/model
+    networks:
+      - sr-fastapi_dev
+    restart: unless-stopped
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: [<gpu_id>]
+              capabilities: [gpu]
+
+networks:
+  sr-fastapi_dev:
+    external: true
\ No newline at end of file