services:
  yue2-test:
    profiles: ["manual"]
    build:
      context: .
      args:
        YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
    image: mike-ai/yue2:3b-0.1.6
    restart: "no"
    environment:
      NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
      NVIDIA_DRIVER_CAPABILITIES: compute,utility
    volumes:
      - /data/models/yue2:/workspace/models:ro
      - /data/music/yue2:/workspace/runs
      - ./requests:/workspace/requests:ro
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
              capabilities: [gpu]