fix: Avoid starting worker if model download RC != 0

Haidra-Org · Jan 23, 2024 · 7b50dba · 7b50dba
1 parent 43bf507
commit 7b50dba
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/horde-bridge.cmd b/horde-bridge.cmd
@@ -1,4 +1,9 @@
 @echo off
 cd /d %~dp0
 call runtime python -s download_models.py
+if ERRORLEVEL 1 GOTO ABORT
+echo "Model Download OK. Starting worker..."
 call runtime python -s run_worker.py %*
+
+:ABORT
+echo "download_models.py exited with error code. Aborting"
diff --git a/horde-bridge.sh b/horde-bridge.sh
@@ -38,5 +38,9 @@ if [ -z "$LD_PRELOAD" ]; then
 fi
 
 
-./runtime.sh python -s download_models.py
-./runtime.sh python -s run_worker.py $*
+if ./runtime.sh python -s download_models.py; then
+    echo "Model Download OK. Starting worker..."
+    ./runtime.sh python -s run_worker.py $*
+else
+    echo "download_models.py exited with error code. Aborting"
+fi