extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-24 17:24:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

Force safetensors 0.3.2 for sdkit, the newer version has issues during installation

Browse Source
This commit is contained in:
cmdr2 2023-08-23 22:04:57 +05:30
parent 15b828b0f5
commit 843ea58c15
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/check_modules.py
View File

@ -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: