From b534c492b8a9cce234dab7f76ba62784e4d90c70 Mon Sep 17 00:00:00 2001
From: nold <nold@gnu.one>
Date: Wed, 21 Feb 2024 16:26:48 +0100
Subject: [PATCH] feat: add runtimeClassName support

---
 charts/local-ai/Chart.yaml                | 2 +-
 charts/local-ai/templates/deployment.yaml | 3 +++
 charts/local-ai/values.yaml               | 3 +++
 3 files changed, 7 insertions(+), 1 deletion(-)

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:
   {}