ACE-Step/docker-compose.yaml at main · JohnnyStreet/ACE-Step · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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