Add safetensors to dependencies

mittagessen · May 10, 2024 · a9dc9e4 · a9dc9e4
1 parent fb773ea
commit a9dc9e4
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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]