diff --git a/clusters/default/media/immich/immich-ml.yml b/clusters/default/media/immich/immich-ml.yml
index 50e6199..fa8b232 100644
--- a/clusters/default/media/immich/immich-ml.yml
+++ b/clusters/default/media/immich/immich-ml.yml
@@ -3,7 +3,7 @@ apiVersion: apps/v1
 kind: Deployment
 metadata:
   name: immich-ml
-  namespace: media
+  namespace: default
 spec:
   strategy:
     type: Recreate
@@ -16,9 +16,10 @@ spec:
       labels:
         app: immich-ml
     spec:
+      runtimeClassName: nvidia
       containers:
       - name: immich-machine-learning
-        image: ghcr.io/immich-app/immich-machine-learning:v2.1.0-openvino
+        image: ghcr.io/immich-app/immich-machine-learning:v2.1.0-cuda
         ports:
         - containerPort: 3003
         env:
@@ -33,9 +34,9 @@ spec:
           mountPath: /cache
         resources:
           requests:
-            gpu.intel.com/i915: "1"
+            nvidia.com/gpu: "1"
           limits:
-            gpu.intel.com/i915: "1"
+            nvidia.com/gpu: "1"
       volumes:
       - name: model-cache
         persistentVolumeClaim: