From a417ebac5bc66e677732bf4886d1d507475115b3 Mon Sep 17 00:00:00 2001
From: Emrul <emrul@emrul.com>
Date: Fri, 7 Mar 2025 14:05:01 +0000
Subject: [PATCH] Update on_sd_start.sh

---
 scripts/on_sd_start.sh | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh
index b55d9072..601a318e 100755
--- a/scripts/on_sd_start.sh
+++ b/scripts/on_sd_start.sh
@@ -48,11 +48,13 @@ if [ -e "ldm" ]; then mv ldm ldm-old; fi
 
 python -m pip install -q torchruntime
 
+cd ..
 # Skip the package download and prompt if INSTALL_ONLY=1 is set
 if [ "$INSTALL_ONLY" != "1" ]; then
-    cd ..
+    # Download the required packages
+    python scripts/check_modules.py
+else
     # Download the required packages
     python scripts/check_modules.py --launch-uvicorn
-
     read -p "Press any key to continue"
 fi