InsightAudio/docker-compose.yml at master · uboy/InsightAudio · 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
71
72
73
74
services:
  insightaudio:
    build:
      context: .
      dockerfile: Dockerfile.gpu
    container_name: insightaudio
    environment:
      - MODEL_DIR=/models
      - CONFIG_DIR=/config
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
      - CELERY_BROKER_URL=redis://redis:6379/0
      - CELERY_RESULT_BACKEND=redis://redis:6379/0
      - LOG_DIR=/app/logs
      - RESULTS_DIR=/app/results
      - MPLCONFIGDIR=/tmp/mplcache
      - APP_UID=1000
      - APP_GID=1000
      - LD_LIBRARY_PATH=/usr/local/lib/python3.10/dist-packages/ctranslate2.libs:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
    ports:
      - "55667:55667"
    volumes:
      - ./models:/models
      - ./config:/config
      - ./tmp:/tmp
      - ./logs:/app/logs
      - ./results:/app/results
    # Поддержка GPU (раскомментируйте если нужен GPU)
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    restart: unless-stopped
  worker:
    build:
      context: .
      dockerfile: Dockerfile.gpu
    container_name: insightaudio-worker
    environment:
      - MODEL_DIR=/models
      - CONFIG_DIR=/config
      - LOG_LEVEL=${LOG_LEVEL:-INFO}
      - CELERY_BROKER_URL=redis://redis:6379/0
      - CELERY_RESULT_BACKEND=redis://redis:6379/0
      - LOG_DIR=/app/logs
      - RESULTS_DIR=/app/results
      - MPLCONFIGDIR=/tmp/mplcache
      - APP_UID=1000
      - APP_GID=1000
      - LD_LIBRARY_PATH=/usr/local/lib/python3.10/dist-packages/ctranslate2.libs:/usr/local/nvidia/lib:/usr/local/nvidia/lib64

    command: celery -A app.celery_app.celery_app worker --loglevel=${LOG_LEVEL:-INFO} -P solo
    volumes:
      - ./models:/models
      - ./config:/config
      - ./tmp:/tmp
      - ./logs:/app/logs
      - ./results:/app/results
    # Поддержка GPU (раскомментируйте если нужен GPU)
    deploy:
       resources:
         reservations:
           devices:
             - driver: nvidia
               count: all
               capabilities: [gpu]
    depends_on:
      - redis
  redis:
    image: redis:7-alpine
    container_name: insightaudio-redis
    restart: unless-stopped