diff --git a/sharktank/requirements.txt b/sharktank/requirements.txt index 19e48f825..6b533d977 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<2.0 # Needed for newer gguf versions (TODO: remove when gguf package includes this) # sentencepiece>=0.1.98,<=0.2.0