Merge pull request #40 from Nold360/feat/runtimeclass

feat: add runtimeClassName support

Author: mudler

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

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

charts/local-ai/values.yaml

@@ -17,6 +17,9 @@ deployment:
   pullPolicy: IfNotPresent
   imagePullSecrets: []
     # - name: secret-names
+  
+  ## Needed for GPU Nodes
+  #runtimeClassName: gpu
 
 resources:
   {}