From 7ae30775c0276e3ad94f3abcef3bd7e07aa3daa0 Mon Sep 17 00:00:00 2001 From: Stephen Baione Date: Fri, 15 Nov 2024 10:26:45 -0600 Subject: [PATCH 1/3] Repin numpy in `sharktank` for `linux` environment --- sharktank/requirements.txt | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sharktank/requirements.txt b/sharktank/requirements.txt index 19e48f825..dd8f14fb6 100644 --- a/sharktank/requirements.txt +++ b/sharktank/requirements.txt @@ -2,8 +2,7 @@ iree-turbine # Runtime deps. gguf==0.6.0 -numpy==1.26.3; sys_platform == 'win32' -numpy; sys_platform != 'win32' +numpy==1.26.3 # Needed for newer gguf versions (TODO: remove when gguf package includes this) # sentencepiece>=0.1.98,<=0.2.0 From 9a49ac241ad562be2910ad9e7a1111020a5e67d8 Mon Sep 17 00:00:00 2001 From: Stephen Baione Date: Fri, 15 Nov 2024 11:24:45 -0600 Subject: [PATCH 2/3] Set numpy version range for windows and non-windows --- sharktank/requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sharktank/requirements.txt b/sharktank/requirements.txt index dd8f14fb6..0e17403ec 100644 --- 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 From 59eff94e86ce71db5cff97832a3318858aa2270f Mon Sep 17 00:00:00 2001 From: Stephen Baione Date: Fri, 15 Nov 2024 11:41:10 -0600 Subject: [PATCH 3/3] Simplify numpy requirement --- sharktank/requirements.txt | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/sharktank/requirements.txt b/sharktank/requirements.txt index 0e17403ec..6b533d977 100644 --- a/sharktank/requirements.txt +++ b/sharktank/requirements.txt @@ -2,8 +2,7 @@ iree-turbine # Runtime deps. gguf==0.6.0 -numpy<2.0; sys_platform == 'win32' -numpy<2.0; sys_platform != 'win32' +numpy<2.0 # Needed for newer gguf versions (TODO: remove when gguf package includes this) # sentencepiece>=0.1.98,<=0.2.0