mirror-secustor-helm-charts/charts/immich/templates/immich-machine-learning.yaml
2025-01-17 22:49:13 +01:00

88 lines
3.1 KiB
YAML

{{- if .Values.machineLearning.enabled -}}
apiVersion: apps/v1
kind: Deployment
metadata:
name: {{ include "immich-machine-learning.name" . }}
labels:
{{- include "immich-machine-learning.labels" . | nindent 4 }}
spec:
replicas: {{ .Values.machineLearning.replicaCount }}
selector:
matchLabels:
{{- include "immich-machine-learning.selectorLabels" . | nindent 6 }}
template:
metadata:
{{- with .Values.machineLearning.podAnnotations }}
annotations:
{{- toYaml . | nindent 8 }}
{{- end }}
labels:
{{- include "immich-machine-learning.labels" . | nindent 8 }}
{{- with .Values.machineLearning.podLabels }}
{{- toYaml . | nindent 8 }}
{{- end }}
spec:
{{- with .Values.imagePullSecrets }}
imagePullSecrets:
{{- toYaml . | nindent 8 }}
{{- end }}
serviceAccountName: {{ include "immich.serviceAccountName" . }}
securityContext:
{{- toYaml .Values.machineLearning.podSecurityContext | nindent 8 }}
containers:
- name: immich-machine-learning
image: "{{ .Values.machineLearning.image.repository }}:{{ .Values.machineLearning.image.tag | default .Chart.AppVersion }}"
imagePullPolicy: {{ .Values.machineLearning.image.pullPolicy }}
ports:
- name: http
containerPort: {{ .Values.machineLearning.service.port }}
protocol: TCP
env:
{{- if .Values.machineLearning.cache.enabled }}
- name: TRANSFORMERS_CACHE_DIR
value: /cache
{{- end }}
{{- if .Values.machineLearning.env }}
{{- toYaml .Values.machineLearning.env | nindent 12 }}
{{- end }}
securityContext:
{{- toYaml .Values.machineLearning.securityContext | nindent 12 }}
livenessProbe:
httpGet:
port: http
path: /ping
resources:
{{- toYaml .Values.machineLearning.resources | nindent 12 }}
volumeMounts:
{{- if .Values.machineLearning.cache.enabled }}
- mountPath: /cache
name: model-cache
{{- end }}
{{- with .Values.machineLearning.volumeMounts }}
{{- toYaml . | nindent 12 }}
{{- end }}
volumes:
{{- if .Values.machineLearning.cache.enabled }}
- name: model-cache
emptyDir:
{{- if .Values.machineLearning.cache.useMemory }}
medium: Memory
{{- end }}
sizeLimit: {{ .Values.machineLearning.cache.sizeLimit }}
{{- end }}
{{- with .Values.machineLearning.volumes }}
{{- toYaml . | nindent 8 }}
{{- end }}
{{- with .Values.machineLearning.nodeSelector }}
nodeSelector:
{{- toYaml . | nindent 8 }}
{{- end }}
{{- with .Values.machineLearning.affinity }}
affinity:
{{- toYaml . | nindent 8 }}
{{- end }}
{{- with .Values.machineLearning.tolerations }}
tolerations:
{{- toYaml . | nindent 8 }}
{{- end }}
{{- end }}