diff --git a/setup.py b/setup.py
index 88ecdea..69c0e39 100644
--- a/setup.py
+++ b/setup.py
@@ -19,6 +19,7 @@ setup(
         "torch",
         "transformers",
         "scipy",
+        "numpy<1.24",
     ],
     packages=find_packages(),
     classifiers=[