From c4b938f13243ead6bb32a47f339b710e2de98cb1 Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Fri, 31 May 2024 10:56:43 +0530
Subject: [PATCH] Hotfix - sdkit's dependencies aren't getting pulled for some
 reason

---
 scripts/check_modules.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/scripts/check_modules.py b/scripts/check_modules.py
index 0861febb..1988bb93 100644
--- a/scripts/check_modules.py
+++ b/scripts/check_modules.py
@@ -137,6 +137,13 @@ def update_modules():
         if accelerate_version < (0, 23):
             install("accelerate", "0.23.0")
 
+    # hotfix - 29 May 2024. sdkit has stopped pulling its dependencies for some reason
+    # temporarily dumping sdkit's requirements here:
+    if os_name != "Windows" and version("picklescan") is None:
+        install_cmd = "python -m pip install gfpgan piexif realesrgan requests picklescan safetensors==0.3.3 k-diffusion==0.0.12 compel==2.0.1 controlnet-aux==0.0.6 invisible-watermark==0.2.0"
+        print(">", install_cmd)
+        os.system(install_cmd)
+
 
 ### utilities