Enha: kokoro to use gpu
This commit is contained in:
@@ -18,20 +18,21 @@ services:
|
|||||||
|
|
||||||
# Kokoro-FastAPI TTS service
|
# Kokoro-FastAPI TTS service
|
||||||
kokoro-tts:
|
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
|
container_name: kokoro-tts
|
||||||
ports:
|
ports:
|
||||||
- "8880:8880"
|
- "8880:8880"
|
||||||
environment:
|
environment:
|
||||||
- API_LOG_LEVEL=INFO
|
- API_LOG_LEVEL=INFO
|
||||||
# For GPU support, uncomment the following lines:
|
# For GPU support, uncomment the following lines:
|
||||||
# deploy:
|
deploy:
|
||||||
# resources:
|
resources:
|
||||||
# reservations:
|
reservations:
|
||||||
# devices:
|
devices:
|
||||||
# - driver: nvidia
|
- driver: nvidia
|
||||||
# count: 1
|
count: 1
|
||||||
# capabilities: [gpu]
|
capabilities: [gpu]
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
|
||||||
volumes:
|
volumes:
|
||||||
|
|||||||
Reference in New Issue
Block a user