Merge pull request #390 from garciadias/master

Author: remsky

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 \

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,19 +70,19 @@ 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]
 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"