fix reqs

segment-any-text · Jun 16, 2024 · 4d222f5 · 4d222f5
1 parent c61973f
commit 4d222f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -10,13 +10,14 @@
     install_requires=[
         "onnxruntime>=1.13.1",
         "transformers>=4.22.2",
-        "numpy>=1",
+        "numpy>=1.0,<=2.0",
         "scikit-learn>=1",
         "tqdm",
         "skops",
         "pandas>=1",
         "cached_property",  # for Py37
         "torchinfo",
+        "fast-mosestokenizer"
     ],
     url="https://github.com/bminixhofer/wtpsplit",
     package_data={"wtpsplit": ["data/*"]},