refactor(charts/cog-ai-model/templates/deployment.yaml,charts/cog-ai-…

…model/values.yaml): Update deployment configuration with runtimeClassName Change the default runtimeClassName in the charts for Cog AI Model. Add a new optional field 'runtimeClassName' to the values.yaml file.
freepik-company · Feb 20, 2024 · 31a606b · 31a606b
1 parent cb86d2f
commit 31a606b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/charts/cog-ai-model/templates/deployment.yaml b/charts/cog-ai-model/templates/deployment.yaml
@@ -23,7 +23,9 @@ spec:
         {{- toYaml . | nindent 8 }}
         {{- end }}
     spec:
-      runtimeClassName: nvidia
+      {{- if not .Values.runtimeClassName }}
+      runtimeClassName: {{ .Values.runtimeClassName }}
+      {{- end }}
       {{- with .Values.imagePullSecrets }}
       imagePullSecrets:
         {{- toYaml . | nindent 8 }}

diff --git a/charts/cog-ai-model/values.yaml b/charts/cog-ai-model/values.yaml
@@ -16,6 +16,8 @@ config:
   modelLocalDir: "/var/huggingface/cache"
   modelMountDir: "/root/.cache/huggingface"
 
+runtimeClassName: nvidia
+
 imagePullSecrets: []
 nameOverride: ""
 fullnameOverride: ""
@@ -153,3 +155,4 @@ readinessProbe:
 
 envs: {}
 envsFrom: {}
+