Enha: kokoro to use gpu

2025-12-06 14:29:46 +03:00
parent ecd7fdee55
commit c49f4e90fe
1 changed files with 10 additions and 9 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -18,20 +18,21 @@ services:

  # Kokoro-FastAPI TTS service
  kokoro-tts:
-    image: ghcr.io/remsky/kokoro-fastapi-cpu:latest
+    # image: ghcr.io/remsky/kokoro-fastapi-cpu:latest
+    image: ghcr.io/remsky/kokoro-fastapi-gpu:latest
    container_name: kokoro-tts
    ports:
      - "8880:8880"
    environment:
      - API_LOG_LEVEL=INFO
    # For GPU support, uncomment the following lines:
-    # deploy:
-    #   resources:
-    #     reservations:
-    #       devices:
-    #         - driver: nvidia
-    #           count: 1
-    #           capabilities: [gpu]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
    restart: unless-stopped

 volumes: