From 30a133bad9206e6646db25292cdaa1629e299728 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Tue, 18 Apr 2023 16:10:46 +0530
Subject: [PATCH] Allow torch 1.11 to continue being installed

---
 scripts/check_modules.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/scripts/check_modules.py b/scripts/check_modules.py
index 60e12fa4..2b47527f 100644
--- a/scripts/check_modules.py
+++ b/scripts/check_modules.py
@@ -1,5 +1,9 @@
 """
 This script checks and installs the required modules.
+
+TODO - Maybe replace the bulk of this script with a call to `pip install -f requirements.txt`, with
+a custom index URL depending on the platform.
+
 """
 
 import os
@@ -9,8 +13,8 @@ import platform
 os_name = platform.system()
 
 modules_to_check = {
-    "torch": ("1.13.1", "2.0.0"),
-    "torchvision": ("0.14.1", "0.15.1"),
+    "torch": ("1.11.0", "1.13.1", "2.0.0"),
+    "torchvision": ("0.12.0", "0.14.1", "0.15.1"),
     "sdkit": "1.0.72",
     "stable-diffusion-sdkit": "2.1.4",
     "rich": "12.6.0",