Dymas/abogen
1
0
Fork 0
mirror of https://github.com/denizsafak/abogen.git synced 2026-07-18 05:40:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: Update Dockerfile and entrypoint script for CUDA diagnostics and web server startup; adjust PyTorch version and URL handling

Browse Source
This commit is contained in:
JB
2025-12-22 08:54:03 -08:00
parent b2058ef3ee
commit 1219b408b5
4 changed files with 53 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker-compose.yaml
+1 -1
View File
@@ -19,7 +19,7 @@ services:
context: .
dockerfile: abogen/webui/Dockerfile
args:
TORCH_INDEX_URL: ${TORCH_INDEX_URL:-https://download.pytorch.org/whl/cu124}
TORCH_INDEX_URL: ${TORCH_INDEX_URL:-https://download.pytorch.org/whl/cu126}
TORCH_VERSION: ${TORCH_VERSION:-}
USE_GPU: ${USE_GPU:-true}
image: abogen:latest