version: "3.8"

services:
  immich-machine-learning:
    image: ghcr.io/immich-app/immich-machine-learning:release-cuda
    container_name: immich-ml
    restart: unless-stopped
    ports:
      - "3003:3003"
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    environment:
      HUGGING_FACE_TOKEN: REDACTED
    volumes:
      - ml-cache:/cache

volumes:
  ml-cache: