{ config, lib, pkgs, ... }:

{
  config = {
    virtualisation.arion.projects.immich-ml.settings = let
      image = { ... }: {
        project.name = "immich-ml";
        services = {
          immich-ml = {
            service = {
              image =
                "ghcr.io/immich-app/immich-machine-learning:${immichVersion}-cuda";
              restart = "always";
              ports = [ "${toString cfg.port}:3003" ];
              volumes = [ "${cfg.state-directory}:/cache" ];
            };
          };
        };
        rawConfig = {
          deploy.resources.reservations.devices = [{
            driver = "nvidia";
            count = 1;
            capabilities = [ "gpu" ];
          }];
        };
      };
    in { imports = [ image ]; };
  };
}