diff --git a/docker/gpu/Dockerfile b/docker/gpu/Dockerfile index 572540cb..0f5ae779 100644 --- a/docker/gpu/Dockerfile +++ b/docker/gpu/Dockerfile @@ -1,4 +1,4 @@ -FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.1-base-ubuntu24.04 +FROM --platform=$BUILDPLATFORM nvcr.io/nvidia/cuda:12.9.1-cudnn-devel-ubuntu24.04 # Install Python and other dependencies RUN apt-get update -y && \ @@ -12,7 +12,7 @@ RUN apt-get update -y && \ useradd -m -u 1001 appuser && \ mkdir -p /app/api/src/models/v1_0 && \ chown -R appuser:appuser /app - + USER appuser WORKDIR /app @@ -40,7 +40,7 @@ ENV PATH="/app/.venv/bin:$PATH" \ PHONEMIZER_ESPEAK_DATA=/usr/share/espeak-ng-data \ ESPEAK_DATA_PATH=/usr/share/espeak-ng-data \ DEVICE="gpu" - + ENV DOWNLOAD_MODEL=true # Download model if enabled RUN if [ "$DOWNLOAD_MODEL" = "true" ]; then \ diff --git a/pyproject.toml b/pyproject.toml index ffbefe54..13447b57 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,8 +32,8 @@ dependencies = [ "mutagen>=1.47.0", "psutil>=6.1.1", "espeakng-loader==0.2.4", - "kokoro==0.9.2", - "misaki[en,ja,ko,zh]==0.9.3", + "kokoro==0.9.4", + "misaki[en,ja,ko,zh]==0.9.4", "spacy==3.8.5", "en-core-web-sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.8.0/en_core_web_sm-3.8.0-py3-none-any.whl", "inflect>=7.5.0", @@ -43,28 +43,19 @@ dependencies = [ ] [project.optional-dependencies] -gpu = [ - "torch==2.7.1+cu128", -] -cpu = [ - "torch==2.7.1", -] +gpu = ["torch==2.8.0+cu129"] +cpu = ["torch==2.8.0"] test = [ "pytest==8.3.5", "pytest-cov==6.0.0", "httpx==0.26.0", "pytest-asyncio==0.25.3", "tomli>=2.0.1", - "jinja2>=3.1.6" + "jinja2>=3.1.6", ] [tool.uv] -conflicts = [ - [ - { extra = "cpu" }, - { extra = "gpu" }, - ], -] +conflicts = [[{ extra = "cpu" }, { extra = "gpu" }]] [tool.uv.sources] torch = [ @@ -79,7 +70,7 @@ explicit = true [[tool.uv.index]] name = "pytorch-cuda" -url = "https://download.pytorch.org/whl/cu128" +url = "https://download.pytorch.org/whl/cu129" explicit = true [build-system] @@ -87,11 +78,11 @@ requires = ["setuptools>=61.0"] build-backend = "setuptools.build_meta" [tool.setuptools] -package-dir = {"" = "api/src"} -packages.find = {where = ["api/src"], namespaces = true} +package-dir = { "" = "api/src" } +packages.find = { where = ["api/src"], namespaces = true } [tool.pytest.ini_options] testpaths = ["api/tests", "ui/tests"] python_files = ["test_*.py"] -addopts = "--cov=api --cov=ui --cov-report=term-missing --cov-config=.coveragerc --full-trace" +addopts = "--cov=api --cov=ui --cov-report=term-missing --cov-config=.coveragerc --full-trace" asyncio_mode = "auto"