Add safetensors dependency, see UKPLab#2372

tomaarsen · Dec 7, 2023 · 0b7b217 · 0b7b217
1 parent 4f75460
commit 0b7b217
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -9,3 +9,4 @@ scipy
 nltk
 sentencepiece
 huggingface-hub
+safetensors
diff --git a/setup.py b/setup.py
@@ -28,7 +28,8 @@
         'scipy',
         'nltk',
         'sentencepiece',
-        'huggingface-hub>=0.4.0'
+        'huggingface-hub>=0.4.0',
+        'safetensors',
     ],
     classifiers=[
         "Development Status :: 5 - Production/Stable",