add dockercopose

2026-03-04 22:39:28 +01:00
parent a86357c190
commit b911db1e07
1 changed files with 23 additions and 0 deletions
--- a/docker-compose.yml
+++ 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: