From 5b5e31233685c1ecb13208c89735e5322f62b8a2 Mon Sep 17 00:00:00 2001
From: Emrul <emrul@emrul.com>
Date: Tue, 11 Mar 2025 09:58:43 +0000
Subject: [PATCH] Update on_sd_start.sh

---
 scripts/on_sd_start.sh | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh
index 08d265c2..ac3201b1 100755
--- a/scripts/on_sd_start.sh
+++ b/scripts/on_sd_start.sh
@@ -62,5 +62,6 @@ else
     # Download the required packages only
     python scripts/check_modules.py
     # Download the models
-    python -c "from easydiffusion.model_manager import init; init()"
+    script_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+    PYTHONPATH="$script_dir/../ui" python -c "from easydiffusion.model_manager import init; init()"
 fi