From 9e7744a9ab08b4362dec8320a3df5ae72c84a319 Mon Sep 17 00:00:00 2001
From: Mert <101130780+mertalev@users.noreply.github.com>
Date: Mon, 31 Mar 2025 19:23:40 -0400
Subject: [PATCH] fix(ml): healthcheck (#17274)

---
 machine-learning/Dockerfile | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile
index bc2fdb88b77..a24f66147f1 100644
--- a/machine-learning/Dockerfile
+++ b/machine-learning/Dockerfile
@@ -67,7 +67,7 @@ FROM python:3.11-slim-bookworm@sha256:7029b00486ac40bed03e36775b864d3f3d39dcbdf1
 
 ENV LD_PRELOAD=/usr/lib/libmimalloc.so.2
 
-FROM prod-cpu AS prod-openvino
+FROM python:3.11-slim-bookworm@sha256:7029b00486ac40bed03e36775b864d3f3d39dcbdf19cd45e6a52d541e6c178f0 AS prod-openvino
 
 RUN apt-get update && \
     apt-get install --no-install-recommends -yqq ocl-icd-libopencl1 wget && \
@@ -82,6 +82,8 @@ RUN apt-get update && \
 
 FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04@sha256:94c1577b2cd9dd6c0312dc04dff9cb2fdce2b268018abc3d7c2dbcacf1155000 AS prod-cuda
 
+ENV LD_PRELOAD=/usr/lib/libmimalloc.so.2
+
 RUN apt-get update && \
     apt-get install --no-install-recommends -yqq libcudnn9-cuda-12 && \
     apt-get clean && \
@@ -140,7 +142,6 @@ ENV TRANSFORMERS_CACHE=/cache \
     PYTHONPATH=/usr/src \
     DEVICE=${DEVICE} \
     VIRTUAL_ENV=/opt/venv \
-    LD_BIND_NOW=1 \
     MACHINE_LEARNING_CACHE_FOLDER=/cache
 
 # prevent core dumps
@@ -149,6 +150,7 @@ RUN echo "hard core 0" >> /etc/security/limits.conf && \
     echo 'ulimit -S -c 0 > /dev/null 2>&1' >> /etc/profile
 
 COPY --from=builder /opt/venv /opt/venv
+COPY scripts/healthcheck.py .
 COPY immich_ml immich_ml
 
 ARG BUILD_ID