Add numpy<2.0 to requirements until tests are sorted out for pytorch 2.3 vs older

2024-06-16 11:24:45 -07:00 · 2024-06-16 11:24:45 -07:00 · 6254dfaece
parent b1a6f4a946
commit 6254dfaece
1 changed files with 2 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -2,4 +2,5 @@ torch>=1.7
 torchvision
 pyyaml
 huggingface_hub
-safetensors>=0.2
+safetensors>=0.2
+numpy<2.0