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"