From dfb004ea46453c1239cd93cb2d2254c5cfdc6156 Mon Sep 17 00:00:00 2001 From: Kjeld Schouten Date: Sat, 8 Mar 2025 21:17:22 +0100 Subject: [PATCH] fix(immicc): allow GPU assignment for ML workload Signed-off-by: Kjeld Schouten --- charts/stable/immich/templates/common.yaml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/charts/stable/immich/templates/common.yaml b/charts/stable/immich/templates/common.yaml index 54a7c6c98ba..8d8129481d6 100644 --- a/charts/stable/immich/templates/common.yaml +++ b/charts/stable/immich/templates/common.yaml @@ -14,7 +14,13 @@ {{/* Enable Machine Learning Service */}} {{- $_ := set .Values.service.machinelearning "enabled" true -}} {{/* Render Machine Learning Workload */}} + {{/* TODO Refactor, tmp fix to allow assigning GPU */}} + {{- $mlResources := .Values.workload.machinelearning.podSpec.containers.machinelearning.resources -}} + {{- $_ := set .Values.workload "machinelearning" (include "immich.machinelearning" . | fromYaml) -}} + {{/* TODO Refactor, tmp fix to allow assigning GPU */}} + {{- $_ := set .Values.workload.machinelearning.podSpec.containers.machinelearning "resources" $mlResources -}} + {{- else -}} {{/* Disable Machine Learning Service */}} {{- $_ := set .Values.service.machinelearning "enabled" false -}}