feat: add Dockerfile and docker-compose.yaml

2025-05-07 09:18:51 +10:00
parent 2143c027b0
commit a1ca0cbf9e
2 changed files with 97 additions and 0 deletions
@@ -0,0 +1,39 @@
+
+services:
+  &name ace-step:
+    build:
+      context: https://github.com/ace-step/ACE-Step.git
+      dockerfile: Dockerfile
+    container_name: *name
+    hostname: *name
+    stop_grace_period: 2s
+    ports:
+      - "7865:7865"
+    volumes:
+      - ace-step-checkpoints:/app/checkpoints
+      - ace-step-outputs:/app/outputs
+      - ace-step-logs:/app/exps/logs
+    # command: python app.py --server_name 0.0.0.0 --port 7865 --share False --bf16 True --torch_compile True --device-id 0
+    healthcheck:
+      test: ["CMD", "curl", "-f", "http://localhost:7865/"]
+      interval: 60s
+      timeout: 10s
+      retries: 30
+      start_period: 3s
+    restart: unless-stopped
+    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: ["compute", "utility", "graphics", "video"]
+
+volumes:
+  ace-step-checkpoints:
+    name: ace-step-checkpoints
+  ace-step-outputs:
+    name: ace-step-outputs
+  ace-step-logs:
+    name: ace-step-logs