vits-simple-api-gsv / docker-compose-gpu.yaml
Artrajz's picture
init
960cd20
version: '3.8'
services:
vits:
image: artrajz/vits-simple-api:latest-gpu
restart: always
ports:
- 23456:23456
environment:
LANG: 'C.UTF-8'
TZ: Asia/Shanghai #timezone
command: [ 'python', 'app.py' ] # 目前用不了gunicorn多进程
volumes:
- ./data:/app/data #挂载模型文件夹
# - ./config.py:/app/config.py # 挂载配置文件
- ./config.yaml:/app/config.yaml # 挂载配置文件
- ./logs:/app/logs # logging logs
- ./gunicorn_config.py:/app/gunicorn_config.py # gunicorn configuration
- ./pyopenjtalk/open_jtalk_dic_utf_8-1.11:/usr/local/lib/python3.10/site-packages/pyopenjtalk/open_jtalk_dic_utf_8-1.11 #pyopentjalk
- ./nltk_data:/usr/local/share/nltk_data
deploy:
resources:
reservations:
devices:
- driver: nvidia
#device_ids: ['0', '3']
capabilities: [ gpu ]