[ci] pin numpy to < 2 on win (#2009)

huggingface · Sep 3, 2024 · d60a1f5 · d60a1f5
1 parent 728a9a3
commit d60a1f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -64,7 +64,8 @@
 REQUIRED_PKGS = [
     "torch>=1.4.0",
     "transformers>=4.31.0",
-    "numpy>=1.18.2",
+    "numpy>=1.18.2;platform_system!='Windows'",
+    "numpy<2;platform_system=='Windows'",
     "accelerate",
     "datasets",
     "tyro>=0.5.11",