From 843ea58c15fbe44a6df5c25d54eaf4fa9f8da5fa Mon Sep 17 00:00:00 2001
From: cmdr2 <secondary.cmdr2@gmail.com>
Date: Wed, 23 Aug 2023 22:04:57 +0530
Subject: [PATCH] Force safetensors 0.3.2 for sdkit, the newer version has
 issues during installation

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

diff --git a/scripts/check_modules.py b/scripts/check_modules.py
index c0c5f6c5..11577500 100644
--- a/scripts/check_modules.py
+++ b/scripts/check_modules.py
@@ -59,6 +59,12 @@ def install(module_name: str, module_version: str):
             module_version = "0.14.1"
 
     install_cmd = f"python -m pip install --upgrade {module_name}=={module_version}"
+
+    # hack for safetensors, until v3 gets released to the main branch
+    if module_name == "sdkit":
+        install_cmd += " safetensors==0.3.2"
+    # /hack
+
     if index_url:
         install_cmd += f" --index-url {index_url}"
     if module_name == "sdkit" and version("sdkit") is not None: