diff --git a/docs/requirements.txt b/docs/requirements.txt
index be87cb592..b101dba18 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,3 +1,4 @@
 myst-parser==0.17.*
 sphinx-rtd-theme==1.0.*
 sphinx==4.5.*
+safetensors
diff --git a/python/tests/requirements.txt b/python/tests/requirements.txt
index c5f4812a7..663819be2 100644
--- a/python/tests/requirements.txt
+++ b/python/tests/requirements.txt
@@ -6,3 +6,4 @@ tensorflow-cpu==2.11.*
 pytest
 wurlitzer==3.0.*;platform_system=='Linux'
 torch==2.2.0
+safetensors