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]

  yue2-ui:
    profiles: ["playground"]
    build:
      context: .
      args:
        YUE2_COMMIT: 9c6c4b349be978b06a9d0d958471a07a6cdeff4d
    image: mike-ai/yue2:3b-0.1.6
    container_name: mike-ai-yue2-playground
    restart: "no"
    entrypoint: ["python", "/opt/yue2-playground/server.py"]
    environment:
      NVIDIA_VISIBLE_DEVICES: "GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"
      NVIDIA_DRIVER_CAPABILITIES: compute,utility
      YUE2_UI_HOST: "0.0.0.0"
      YUE2_UI_PORT: "8014"
    ports:
      - "127.0.0.1:8014:8014"
    volumes:
      - /data/models/yue2:/workspace/models:ro
      - /data/music/yue2:/workspace/runs
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              device_ids: ["GPU-8ad38c6c-5a01-9d8e-1dfa-ed662ad78fbe"]
              capabilities: [gpu]
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://127.0.0.1:8014/health', timeout=2)"]
      interval: 5s
      timeout: 3s
      retries: 12
      start_period: 5s