version: '3.8'

services:
  cosyvoice:
    build: .
    image: cosyvoice:latest
    ports:
      - "50000:50000"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]
    volumes:
      - ./pretrained_models:/app/pretrained_models
    command: python3 app.py --port 50000 --model_dir pretrained_models/CosyVoice-300M