From 2c4452f5d132198ed381a7b262b4a5cab5114b5f Mon Sep 17 00:00:00 2001
From: mertalev <101130780+mertalev@users.noreply.github.com>
Date: Wed, 5 Mar 2025 19:20:39 -0500
Subject: [PATCH] increase timeout for rocm

---
 machine-learning/start.sh | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/machine-learning/start.sh b/machine-learning/start.sh
index 859183851c..0a1aa4e9bd 100755
--- a/machine-learning/start.sh
+++ b/machine-learning/start.sh
@@ -2,17 +2,14 @@
 
 echo "Initializing Immich ML $IMMICH_SOURCE_REF"
 
-if ! [ "$DEVICE" = "openvino" ]; then
-	: "${MACHINE_LEARNING_WORKER_TIMEOUT:=120}"
-else
-	: "${MACHINE_LEARNING_WORKER_TIMEOUT:=300}"
-fi
-
 # mimalloc seems to increase memory usage dramatically with openvino, need to investigate
 if ! [ "$DEVICE" = "openvino" ] && ! [ "$DEVICE" = "rocm" ]; then
 	lib_path="/usr/lib/$(arch)-linux-gnu/libmimalloc.so.2"
 	export LD_PRELOAD="$lib_path"
 	export LD_BIND_NOW=1
+	: "${MACHINE_LEARNING_WORKER_TIMEOUT:=120}"
+else
+	: "${MACHINE_LEARNING_WORKER_TIMEOUT:=300}"
 fi
 
 : "${IMMICH_HOST:=[::]}"