Try newer version of numpy

pytorch · Aug 19, 2024 · 8355552 · 8355552
1 parent a974a31
commit 8355552
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torch_ort/requirements-dev.txt b/torch_ort/requirements-dev.txt
@@ -1,7 +1,8 @@
 # required for CI test
 pandas
 scikit-learn
-numpy
+numpy==2.1.0 ; python_version < '3.11'
+numpy==2.1.0 ; python_version >= '3.11'
 transformers==v4.4.2
 torch
 torchvision