diff --git a/packages/text-embeddings/chart/templates/deployment.yaml b/packages/text-embeddings/chart/templates/deployment.yaml index 8f4adde05..c83656d4e 100644 --- a/packages/text-embeddings/chart/templates/deployment.yaml +++ b/packages/text-embeddings/chart/templates/deployment.yaml @@ -30,9 +30,9 @@ spec: {{- $gpuLimit = $gpuLimit | float64 }} {{- end }} {{- if gt $gpuLimit 0.0 }} - runtimeClassName: nvidia + runtimeClassName: nvidia {{- else if .Values.gpu.runtimeClassName }} - runtimeClassName: {{ .Values.gpu.runtimeClassName }} + runtimeClassName: {{ .Values.gpu.runtimeClassName }} {{- end }} # It's necessary to include the ###ZARF_DATA_INJECTION_MARKER### somewhere in the podspec, otherwise data injections will not occur. initContainers: diff --git a/packages/whisper/chart/templates/deployment.yaml b/packages/whisper/chart/templates/deployment.yaml index e9a4df2c9..274914205 100644 --- a/packages/whisper/chart/templates/deployment.yaml +++ b/packages/whisper/chart/templates/deployment.yaml @@ -30,9 +30,9 @@ spec: {{- $gpuLimit = $gpuLimit | float64 }} {{- end }} {{- if gt $gpuLimit 0.0 }} - runtimeClassName: nvidia + runtimeClassName: nvidia {{- else if .Values.gpu.runtimeClassName }} - runtimeClassName: {{ .Values.gpu.runtimeClassName }} + runtimeClassName: {{ .Values.gpu.runtimeClassName }} {{- end }} # It's necessary to include the ###ZARF_DATA_INJECTION_MARKER### somewhere in the podspec, otherwise data injections will not occur. initContainers: