mirror of
https://github.com/babysor/MockingBird.git
synced 2024-03-22 13:11:31 +08:00
24 lines
566 B
YAML
24 lines
566 B
YAML
|
version: '3.8'
|
||
|
|
|
||
|
|
services:
|
||
|
|
server:
|
||
|
|
image: mockingbird:latest
|
||
|
|
build: .
|
||
|
|
volumes:
|
||
|
|
- ./datasets:/datasets
|
||
|
|
- ./synthesizer/saved_models:/workspace/synthesizer/saved_models
|
||
|
|
environment:
|
||
|
|
- DATASET_MIRROR=US
|
||
|
|
- FORCE_RETRAIN=false
|
||
|
|
- TRAIN_DATASETS=aidatatang_200zh magicdata aishell3 data_aishell
|
||
|
|
- TRAIN_SKIP_EXISTING=true
|
||
|
|
ports:
|
||
|
|
- 8080:8080
|
||
|
|
deploy:
|
||
|
|
resources:
|
||
|
|
reservations:
|
||
|
|
devices:
|
||
|
|
- driver: nvidia
|
||
|
|
device_ids: [ '0' ]
|
||
|
|
capabilities: [ gpu ]
|