docker-compose.gpu.yml

Rama main · réplica de lectura del repositorio original.

# NVIDIA GPU override
#
# Usage:
#   docker compose -f docker-compose.yml -f docker-compose.gpu.yml up --build
#
# Env vars and ENABLE_WHISPER_DIARIZE are inherited from the base
# compose file — this override only flips GPU_RUNTIME and attaches
# the GPU device. On GPU, pyannote.audio diarization runs ~10-15×
# faster than CPU, so enabling ENABLE_WHISPER_DIARIZE=1 here is
# significantly more practical than on the CPU image.

name: clippyme

services:
  backend:
    build:
      args:
        GPU_RUNTIME: nvidia
        # Inherit user preference; default 0 if not set.
        ENABLE_WHISPER_DIARIZE: ${ENABLE_WHISPER_DIARIZE:-0}
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]