Remove safetensors hack (#1525)

We should upgrade to 0.3.3, since it has wheels for all our supported plattforms.
This commit is contained in:
JeLuF 2023-08-29 05:50:32 +02:00 committed by cmdr2
parent 80c4a50ca1
commit ad227ca190

View File

@ -60,11 +60,6 @@ def install(module_name: str, module_version: str):
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: