diff --git a/setup.py b/setup.py index 6d7a59ad0f..542765c373 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,!=2.0.0", + "numpy>=1.18.2", "accelerate", "datasets", "tyro>=0.5.11",