version: "3"
services:
  vits:
    image: vits-moe:latest
    restart: always
    ports:
      - "7860:7860"
    environment:
      - TZ=Asia/Shanghai
    volumes:
      - ./config:/app/config
      - ./pretrained_models:/app/pretrained_models
      - ./config.py:/app/config.py
      - ./cuda_test.py:/app/cuda_test.py
      - ./entrypoint.sh:/app/entrypoint.sh
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: 1
              capabilities: [gpu]