From 2471d9ba21462dd43624e2522c00fac762f70a53 Mon Sep 17 00:00:00 2001
From: Emrul <emrul@emrul.com>
Date: Tue, 11 Mar 2025 03:53:41 +0000
Subject: [PATCH] Update on_sd_start.sh

Fix for weight load error
---
 scripts/on_sd_start.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh
index e5a401c2..ea6f33b6 100755
--- a/scripts/on_sd_start.sh
+++ b/scripts/on_sd_start.sh
@@ -52,6 +52,10 @@ cd ..
 # Skip the package download and prompt if INSTALL_ONLY=1 is set
 if [ "$INSTALL_ONLY" != "1" ]; then
     # Download the required packages
+
+    # see https://github.com/easydiffusion/easydiffusion/issues/1911
+    export TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD=1
+    
     python scripts/check_modules.py --launch-uvicorn
     read -p "Press any key to continue"
 else