Subgen-English-Plex/docker-compose.gpu.yml at main · Herbertmt978/Subgen-English-Plex · 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
services:
  subgen:
    container_name: subgen
    image: ${SUBGEN_IMAGE:-ghcr.io/herbertmt978/subgen-english-plex:v0.3.0}
    restart: unless-stopped
    mem_limit: ${SUBGEN_MEMORY_LIMIT:-10g}
    memswap_limit: ${SUBGEN_MEMORY_LIMIT:-10g}
    pids_limit: 512
    cpus: ${SUBGEN_CPU_LIMIT:-4.0}
    gpus: all
    ports:
      - "${SUBGEN_BIND_ADDRESS:-127.0.0.1}:9000:9000"
    environment:
      - PUID=${PUID:-1000}
      - PGID=${PGID:-1000}
      - WEBHOOK_PORT=9000
      - SUBGEN_API_KEY=${SUBGEN_API_KEY:-}
      - HTTP_TIMEOUT_SECONDS=30
      - TRANSCRIBE_DEVICE=cuda
      - WHISPER_MODEL=${WHISPER_MODEL:-medium}
      - COMPUTE_TYPE=float16
      - CONCURRENT_TRANSCRIPTIONS=1
      - WHISPER_THREADS=${WHISPER_THREADS:-4}
      - TRANSCRIBE_FOLDERS=${TRANSCRIBE_FOLDERS:-/media/Movies|/media/TV}
      - MONITOR=True
      - PROCESS_ADDED_MEDIA=True
      - PROCESS_MEDIA_ON_PLAY=False
      - MODEL_PATH=/subgen/models
      - TRANSCRIBE_OR_TRANSLATE=translate
      - SUBTITLE_LANGUAGE_NAME=en
      - SUBTITLE_LANGUAGE_NAMING_TYPE=ISO_639_1
      - SHOULD_WHISPER_DETECT_AUDIO_LANGUAGE=True
      - NOTIFY_ON_ENGLISH_AUDIO_MISMATCH=True
      - SKIP_IF_TARGET_SUBTITLES_EXIST=True
      - SKIP_IF_EXTERNAL_SUBTITLES_EXIST=True
      - CLEAR_VRAM_ON_COMPLETE=True
      - MODEL_CLEANUP_DELAY=${MODEL_CLEANUP_DELAY:-300}
      - SKIP_VIDEO_EXTENSIONS=.avi
      - PLEX_SERVER=${PLEX_SERVER:-}
      - PLEX_TOKEN=${PLEX_TOKEN:-}
      - USE_PATH_MAPPING=${USE_PATH_MAPPING:-False}
      - PATH_MAPPING_FROM=${PATH_MAPPING_FROM:-}
      - PATH_MAPPING_TO=${PATH_MAPPING_TO:-}
      - NVIDIA_VISIBLE_DEVICES=all
      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
      - DEBUG=False
    volumes:
      - ${MEDIA_ROOT:-/srv/media}:/media
      - ${SUBGEN_MODEL_PATH:-./models}:/subgen/models
    stop_grace_period: 2m