From ccf25cc99cd95cfc0b8ef5cb1a3dd7a5bc754da2 Mon Sep 17 00:00:00 2001
From: Viktor Barzin <viktorbarzin@meta.com>
Date: Fri, 6 Feb 2026 20:28:29 +0000
Subject: [PATCH] Upgrade immich to v2.5.2 and add GPU toleration to ML pod

Bump immich version from v2.5.0 to v2.5.2. Add nvidia.com/gpu
toleration to immich-machine-learning deployment.
---
 modules/kubernetes/immich/main.tf | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/modules/kubernetes/immich/main.tf b/modules/kubernetes/immich/main.tf
index 1a8988e9..3a833362 100644
--- a/modules/kubernetes/immich/main.tf
+++ b/modules/kubernetes/immich/main.tf
@@ -5,7 +5,7 @@ variable "homepage_token" {}
 variable "immich_version" {
   type = string
   # Change me to upgrade
-  default = "v2.5.0"
+  default = "v2.5.2"
 }
 
 
@@ -363,6 +363,12 @@ resource "kubernetes_deployment" "immich-machine-learning" {
         node_selector = {
           "gpu" : "true"
         }
+        toleration {
+          key      = "nvidia.com/gpu"
+          operator = "Equal"
+          value    = "true"
+          effect   = "NoSchedule"
+        }
         container {
           # image = "ghcr.io/immich-app/immich-machine-learning:${var.immich_version}"
           image = "ghcr.io/immich-app/immich-machine-learning:${var.immich_version}-cuda"