diff --git a/setup.py b/setup.py
index 6ad31563f28..65ae9293d57 100644
--- a/setup.py
+++ b/setup.py
@@ -63,7 +63,7 @@
 REQUIRED_PKGS = [
     "torch>=1.4.0",
     "transformers>=4.31.0",
-    "numpy>=1.18.2",
+    "numpy>=1.18.2,<2.0.0",
     "accelerate",
     "datasets",
     "tyro>=0.5.11",