vits-simple-api / docker-compose-gpu.yaml
Artrajz's picture
update
b0f5083
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:
- ./Model:/app/Model # 挂载模型文件夹
- ./config.py:/app/config.py # 挂载配置文件
- ./logs:/app/logs # logging logs
- ./gunicorn_config.py:/app/gunicorn_config.py # gunicorn configuration
- ./vits/bert:/app/vits/bert # vits_chinese
- ./bert_vits2/bert/chinese-roberta-wwm-ext-large:/app/bert_vits2/bert/chinese-roberta-wwm-ext-large # Bert-vits2 chinese
- ./bert_vits2/bert/bert-base-japanese-v3:/app/bert_vits2/bert/bert-base-japanese-v3 # Bert-vits2 japanese
- ./pyopenjtalk/open_jtalk_dic_utf_8-1.11:/usr/local/lib/python3.10/site-packages/pyopenjtalk/open_jtalk_dic_utf_8-1.11 #pyopentjalk
deploy:
resources:
reservations:
devices:
- driver: nvidia
#device_ids: ['0', '3']
capabilities: [gpu]