LocalImageGen/compose.yaml at master · RCellar/LocalImageGen · 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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# compose.yaml — works with podman-compose and docker-compose
# Generated .env file provides: MODELS_DIR, OUTPUTS_DIR, INVOKEAI_PORT, COGVIDEO_PORT, GPU_DEVICE

services:
  invokeai:
    image: ghcr.io/invoke-ai/invokeai
    profiles: ["image", "all"]
    ports:
      - "${INVOKEAI_PORT:-9090}:${INVOKEAI_PORT:-9090}"
    volumes:
      - ${MODELS_DIR:-./models}:/models
      - invokeai-data:/invokeai
      - ./containers/invokeai/entrypoint.sh:/entrypoint-wrapper.sh:ro
    environment:
      - INVOKEAI_PORT=${INVOKEAI_PORT:-9090}
      - INVOKEAI_ROOT=/invokeai
      - NVIDIA_VISIBLE_DEVICES=${GPU_DEVICE:-0}
    entrypoint: ["/bin/bash", "/entrypoint-wrapper.sh"]
    security_opt:
      - label=disable
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:${INVOKEAI_PORT:-9090}/api/v1/app/version"]
      interval: 30s
      timeout: 10s
      start_period: 120s
      retries: 3
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

  cogvideo:
    build:
      context: ./containers/cogvideo
      dockerfile: Containerfile
    image: cogvideo-local:latest
    profiles: ["video", "all"]
    ports:
      - "${COGVIDEO_PORT:-7860}:${COGVIDEO_PORT:-7860}"
    volumes:
      - ${MODELS_DIR:-./models}:/models:ro
      - ${OUTPUTS_DIR:-./outputs}/videos:/outputs/videos
    environment:
      - MODEL_PATH=/models/cogvideox-5b
      - I2V_MODEL_PATH=/models/cogvideox-5b-i2v
      - OUTPUT_DIR=/outputs/videos
      - COGVIDEO_PORT=${COGVIDEO_PORT:-7860}
      - COGVIDEO_QUANTIZATION=${COGVIDEO_QUANTIZATION:-none}
      - NVIDIA_VISIBLE_DEVICES=${GPU_DEVICE:-0}
    security_opt:
      - label=disable
    healthcheck:
      test: ["CMD", "curl", "-f", "http://localhost:${COGVIDEO_PORT:-7860}/"]
      interval: 30s
      timeout: 10s
      start_period: 120s
      retries: 3
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]

volumes:
  invokeai-data: