From 03c49d5b47f4ec651c24483ce5938894d78e3090 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Mon, 10 Feb 2025 19:31:05 +0530
Subject: [PATCH] Hotfix for missing torchruntime on new installations

---
 scripts/on_sd_start.bat | 2 ++
 scripts/on_sd_start.sh  | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/scripts/on_sd_start.bat b/scripts/on_sd_start.bat
index 2d5ff0ba..bee3b0db 100644
--- a/scripts/on_sd_start.bat
+++ b/scripts/on_sd_start.bat
@@ -71,6 +71,8 @@ echo PYTHONPATH=%PYTHONPATH%
 call where python
 call python --version
 
+call python -m pip install -q torchruntime
+
 call python scripts\check_modules.py --launch-uvicorn
 pause
 exit /b
diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh
index a9fc809d..0e9f1a08 100755
--- a/scripts/on_sd_start.sh
+++ b/scripts/on_sd_start.sh
@@ -50,6 +50,8 @@ fi
 if [ -e "src" ]; then mv src src-old; fi
 if [ -e "ldm" ]; then mv ldm ldm-old; fi
 
+python -m pip install -q torchruntime
+
 cd ..
 # Download the required packages
 python scripts/check_modules.py --launch-uvicorn