diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..3ef3de1
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,23 @@
+version: '3.8'
+
+services:
+  dia-api:
+    build: .
+    container_name: dia-api-server
+    ports:
+      - "8000:8000"
+    volumes:
+      - hf_cache:/root/.cache/huggingface
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+    environment:
+      - MODEL_ID=nari-labs/Dia-1.6B
+      - PORT=8000
+
+volumes:
+  hf_cache: