Add updates to torch tags here as well

This commit is contained in:
Drew Thomasson
2025-06-01 20:40:17 -04:00
committed by GitHub
parent 0e2406a820
commit 30c5118bb5

View File

@@ -13,7 +13,7 @@ services:
build:
context: .
args:
TORCH_VERSION: cuda11 # Available tags: [cuda12, cuda11, cuda128, rocm, xpu, cpu]
TORCH_VERSION: cuda118 # Available tags: [cuda121, cuda118, cuda128, rocm, xpu, cpu] # All CUDA version numbers should work, Ex: CUDA 11.6-> cuda116
SKIP_XTTS_TEST: "true" # (Saves space by not baking xtts model into docker image)
# To update ebook2audiobook to the latest you may have to rebuild
entrypoint: ["python", "app.py", "--script_mode", "full_docker"]