add support for GPU (cuda)

This commit is contained in:
pluja
2023-10-27 10:23:08 +02:00
parent f4484884cc
commit 5570a745f8
7 changed files with 170 additions and 22 deletions

View File

@@ -17,10 +17,9 @@ services:
command: ['--logpath', '/var/log/mongodb/mongod.log']
healthcheck:
test: mongosh --eval 'db.runCommand("ping").ok' --quiet
interval: 15s
timeout: 10s
interval: 2s
timeout: 3s
retries: 5
start_period: 5s
translate:
container_name: whisper-libretranslate
@@ -39,9 +38,8 @@ services:
- 5000
healthcheck:
test: ['CMD-SHELL', './venv/bin/python scripts/healthcheck.py']
interval: 15s
timeout: 10s
start_period: 5s
interval: 2s
timeout: 3s
retries: 5
transcription-api:
@@ -76,10 +74,9 @@ services:
condition: service_healthy
healthcheck:
test: curl --fail http://localhost:8080/api/transcriptions || exit 1
interval: 15s
timeout: 10s
interval: 2s
timeout: 2s
retries: 5
start_period: 5s
expose:
- 8080
@@ -90,9 +87,10 @@ services:
- .env
restart: unless-stopped
environment:
INTERNAL_API_HOST: "http://nginx:80"
VITE_TRANSLATION_API_HOST: ""
VITE_API_HOST: ${WHISHPER_HOST:-}
PUBLIC_INTERNAL_API_HOST: "http://nginx:80"
PUBLIC_TRANSLATION_API_HOST: ""
PUBLIC_API_HOST: ${WHISHPER_HOST:-}
PUBLIC_WHISHPER_GPU: ${WHISHPER_GPU:-false}
depends_on:
transcription-api:
condition: service_healthy