diff options
| author | Grail Finder <wohilas@gmail.com> | 2025-12-06 14:29:46 +0300 |
|---|---|---|
| committer | Grail Finder <wohilas@gmail.com> | 2025-12-07 13:22:07 +0300 |
| commit | c49f4e90fed02122dc6dbf4837b584b7207e073a (patch) | |
| tree | 5d1f91d8dad0a35e6636def0f489cb18e911d9b8 | |
| parent | ecd7fdee551cb43d17a79bea1f76cc263598d58c (diff) | |
Enha: kokoro to use gpu
| -rw-r--r-- | docker-compose.yml | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/docker-compose.yml b/docker-compose.yml index 81a6d9d..c014814 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -18,24 +18,25 @@ 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: models: driver: local audio: - driver: local
\ No newline at end of file + driver: local |