From 4bd41ccf775856f318f33a5682f1561ab0f66d20 Mon Sep 17 00:00:00 2001
From: Daniel Cardozo <daniel.cardozo@zweiloewen.com>
Date: Tue, 28 Oct 2025 12:56:01 +0100
Subject: [PATCH] Added configs to enable machine learning gpu feature

---
 .../apps/immich/machine-learning/DaemonSet.yaml  | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/cluster/apps/immich/machine-learning/DaemonSet.yaml b/cluster/apps/immich/machine-learning/DaemonSet.yaml
index bde9116..f8c2c4d 100644
--- a/cluster/apps/immich/machine-learning/DaemonSet.yaml
+++ b/cluster/apps/immich/machine-learning/DaemonSet.yaml
@@ -4,7 +4,7 @@ metadata:
   name: immich-machine-learning
   namespace: immich
   annotations:
-    reloader.stakater.com/auto: 'true'
+    reloader.stakater.com/auto: "true"
   labels:
     app: immich-machine-learning
 spec:
@@ -19,7 +19,7 @@ spec:
       automountServiceAccountToken: false
       containers:
         - name: immich-machine-learning
-          image: ghcr.io/immich-app/immich-machine-learning:v2.1.0 # {"$imagepolicy": "immich:immich-machine-learning"}
+          image: ghcr.io/immich-app/immich-machine-learning:v2.1.0-openvino # {"$imagepolicy": "immich:immich-machine-learning"}
           imagePullPolicy: IfNotPresent
           resources:
             limits:
@@ -52,6 +52,14 @@ spec:
                   name: immich-db-app
                   key: password
                   optional: false
+            # Check this ENVs maybe it works also for other ML models
+            - name: MACHINE_LEARNING_PRELOAD__CLIP
+              value: ViT-B-32__openai
+            - name: MACHINE_LEARNING_PRELOAD__FACIAL_RECOGNITION
+              value: buffalo_l
+            # May not be needed
+            - name: MACHINE_LEARNING_EXECUTION_PROVIDERS
+              value: "OpenVINOExecutionProvider,CPUExecutionProvider"
           envFrom:
             - configMapRef:
                 name: immich
@@ -72,12 +80,12 @@ spec:
           securityContext:
             privileged: true # for intel quick sync
             seccompProfile:
-              type: 'RuntimeDefault'
+              type: "RuntimeDefault"
           volumeMounts:
             - name: immich-machine-learning
               mountPath: /cache
       nodeSelector:
-        intel.feature.node.kubernetes.io/gpu: 'true'
+        intel.feature.node.kubernetes.io/gpu: "true"
       volumes:
         - name: immich-machine-learning
           persistentVolumeClaim: