feat: add runtimeClassName support

go-skynet · Feb 21, 2024 · b534c49 · b534c49
1 parent 6a3e8b1
commit b534c49
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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
@@ -17,6 +17,9 @@ deployment:
   pullPolicy: IfNotPresent
   imagePullSecrets: []
     # - name: secret-names
+
+  ## Needed for GPU Nodes
+  #runtimeClassName: gpu
 
 resources:
   {}