Set numpy version range for windows and non-windows

nod-ai · Nov 15, 2024 · 9a49ac2 · 9a49ac2
1 parent 7ae3077
commit 9a49ac2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sharktank/requirements.txt b/sharktank/requirements.txt
@@ -2,7 +2,8 @@ iree-turbine
 
 # Runtime deps.
 gguf==0.6.0
-numpy==1.26.3
+numpy<2.0; sys_platform == 'win32'
+numpy<2.0; sys_platform != 'win32'
 
 # Needed for newer gguf versions (TODO: remove when gguf package includes this)
 # sentencepiece>=0.1.98,<=0.2.0