diff --git a/charts/local-ai/Chart.yaml b/charts/local-ai/Chart.yaml
index 13cde4c..f2d04a9 100644
--- a/charts/local-ai/Chart.yaml
+++ b/charts/local-ai/Chart.yaml
@@ -3,4 +3,4 @@ appVersion: 1.40
 description: A Helm chart for deploying LocalAI to a Kubernetes cluster
 name: local-ai
 type: application
-version: 3.1.0
+version: 3.2.0
diff --git a/charts/local-ai/templates/deployment.yaml b/charts/local-ai/templates/deployment.yaml
index 40a946b..d8dcbe4 100644
--- a/charts/local-ai/templates/deployment.yaml
+++ b/charts/local-ai/templates/deployment.yaml
@@ -29,6 +29,9 @@ spec:
         checksum/config-prompt-templates: {{ include (print $.Template.BasePath "/configmap-prompt-templates.yaml") . | sha256sum }}
         {{- end }}
     spec:
+      {{- with .Values.deployment.runtimeClassName }}
+      runtimeClassName: {{ . }}
+      {{- end }}
       {{- with .Values.deployment.imagePullSecrets }}
       imagePullSecrets:
         {{- toYaml . | nindent 8 }}
diff --git a/charts/local-ai/values.yaml b/charts/local-ai/values.yaml
index f446175..b1dbbe1 100644
--- a/charts/local-ai/values.yaml
+++ b/charts/local-ai/values.yaml
@@ -17,6 +17,9 @@ deployment:
   pullPolicy: IfNotPresent
   imagePullSecrets: []
     # - name: secret-names
+  
+  ## Needed for GPU Nodes
+  #runtimeClassName: gpu
 
 resources:
   {}