Specify torch version of safetensors

OpenNMT · Dec 5, 2024 · 488e362 · 488e362
1 parent 687147c
commit 488e362
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,4 +1,4 @@
 myst-parser==0.17.*
 sphinx-rtd-theme==1.0.*
 sphinx==4.5.*
-safetensors
+safetensors[torch]