From 5af0b2faf77751593ff82718bafd7af7fea88314 Mon Sep 17 00:00:00 2001
From: niten <niten@fudo.org>
Date: Mon, 20 May 2024 16:51:58 -0700
Subject: [PATCH] Disable CPU ML by default

---
 immich-container.nix | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/immich-container.nix b/immich-container.nix
index 906e743..86ddc46 100644
--- a/immich-container.nix
+++ b/immich-container.nix
@@ -19,6 +19,9 @@ in {
     enable =
       mkEnableOption "Enable Immich photo server running in a container.";
 
+    cpu-machine-learning =
+      mkEnableOption "Perform machine learning using the local CPU.";
+
     state-directory = mkOption {
       type = str;
       description = "Path at which to store server state.";
@@ -115,7 +118,7 @@ in {
             };
           };
 
-          immich-machine-learning = {
+          immich-machine-learning = mkIf cfg.cpu-machine-learning {
             service = {
               image = cfg.images.immich-ml;
               restart = "always";