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]