From a9dc9e4f9ca37c90e638acd9d52d21953912875f Mon Sep 17 00:00:00 2001 From: Benjamin Kiessling Date: Fri, 10 May 2024 17:38:09 +0200 Subject: [PATCH] Add safetensors to dependencies --- conda/recipe.yaml | 1 + environment.yml | 1 + environment_cuda.yml | 1 + setup.cfg | 1 + 4 files changed, 4 insertions(+) diff --git a/conda/recipe.yaml b/conda/recipe.yaml index 18128d46d..b8a669793 100644 --- a/conda/recipe.yaml +++ b/conda/recipe.yaml @@ -44,6 +44,7 @@ requirements: - lightning~=2.2 - torchmetrics>=1.1.0 - conda-forge::threadpoolctl~=3.4.0 + - conda-forge::safetensors~=0.4.0 - albumentations - rich diff --git a/environment.yml b/environment.yml index 242b8426c..b4614e25d 100644 --- a/environment.yml +++ b/environment.yml @@ -26,6 +26,7 @@ dependencies: - conda-forge::lightning~=2.2.0 - conda-forge::torchmetrics>=1.1.0 - conda-forge::threadpoolctl~=3.4 + - conda-forge::safetensors~=0.4.0 - pip - albumentations - rich diff --git a/environment_cuda.yml b/environment_cuda.yml index 83b75f850..e649eff28 100644 --- a/environment_cuda.yml +++ b/environment_cuda.yml @@ -27,6 +27,7 @@ dependencies: - conda-forge::lightning~=2.2.0 - conda-forge::torchmetrics>=1.1.0 - conda-forge::threadpoolctl~=3.4 + - conda-forge::safetensors~=0.4.0 - pip - albumentations - rich diff --git a/setup.cfg b/setup.cfg index 5f675f35e..f66db8ff6 100644 --- a/setup.cfg +++ b/setup.cfg @@ -62,6 +62,7 @@ install_requires = torchmetrics>=1.1.0 threadpoolctl~=3.4.0 importlib-resources>=1.3.0 + safetensors~=0.4.0 rich [options.extras_require]