diff --git a/setup.py b/setup.py
index fe42e8833f0..fb7d133f111 100644
--- a/setup.py
+++ b/setup.py
@@ -76,7 +76,7 @@
         "pyyaml",
         "torch>=1.10.0",
         "huggingface_hub>=0.21.0",
-        "safetensors>=0.3.1",
+        "safetensors>=0.4.3",
     ],
     extras_require=extras,
     classifiers=[