25 lines
767 B
YAML
25 lines
767 B
YAML
services:
|
|
api-chat:
|
|
image: api_chat:latest
|
|
volumes:
|
|
- /dockerdata/models:/obscura/models
|
|
- /dockerdata/task/audio_files:/obscura/task/audio_files
|
|
- /dockerdata/api_chat/TTS:/api_chat/TTS
|
|
- /dockerdata/api_chat/sample:/api_chat/sample
|
|
- /dockerdata/api_chat/.env:/api_chat/.env
|
|
- /dockerdata/api_chat/logs:/api_chat/logs
|
|
- /dockerdata/api_chat/pids:/api_chat/pids
|
|
- /dockerdata/nltk_data:/root/nltk_data
|
|
environment:
|
|
- TZ=Asia/Shanghai
|
|
- CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
|
|
- NLTK_DATA=/root/nltk_data
|
|
network_mode: "host"
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: [gpu]
|