Enha: kokoro to use gpu
This commit is contained in:
@@ -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
|
||||
driver: local
|
||||
|
||||
Reference in New Issue
Block a user