From 6254dfaece48589d3865d2bfda7e37b80bbc3b68 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 16 Jun 2024 11:24:45 -0700 Subject: [PATCH] Add numpy<2.0 to requirements until tests are sorted out for pytorch 2.3 vs older --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index b750d03d5e..918fda0ec4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,4 +2,5 @@ torch>=1.7 torchvision pyyaml huggingface_hub -safetensors>=0.2 \ No newline at end of file +safetensors>=0.2 +numpy<2.0