feat: Add USE_GPU argument to Dockerfile and docker-compose for GPU support

2026-07-18 13:40:27 +02:00 · 2025-12-21 08:31:51 -08:00
parent 938e122166
commit 899c9f5aa5
2 changed files with 8 additions and 0 deletions
@@ -16,6 +16,7 @@ services:
      args:
        TORCH_INDEX_URL: ${TORCH_INDEX_URL:-https://download.pytorch.org/whl/cu124}
        TORCH_VERSION: ${TORCH_VERSION:-}
+        USE_GPU: ${USE_GPU:-true}
    image: abogen:latest
    user: "${ABOGEN_UID:-1000}:${ABOGEN_GID:-1000}"
    ports: