From a9da65c2cd2d3e6b20f1645722a115c94fd7ae53 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Fri, 17 Mar 2023 13:27:44 +0530
Subject: [PATCH] Temp fix for macOS installations that installed torch 2.0 by
 mistake, when torch 2 released yesterday

---
 scripts/on_sd_start.sh | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh
index 4b1b7ab5..b9bc7950 100755
--- a/scripts/on_sd_start.sh
+++ b/scripts/on_sd_start.sh
@@ -65,6 +65,11 @@ esac
 
 # install torch and torchvision
 if python ../scripts/check_modules.py torch torchvision; then
+    # temp fix for macOS installations that installed torch 2.0 by mistake
+    if [ "$OS_NAME" == "macos" ]; then
+        python -m pip install --upgrade torch==1.13.1 torchvision==0.14.1 -q
+    fi
+
     echo "torch and torchvision have already been installed."
 else
     echo "Installing torch and torchvision.."