diff --git a/requirements-rocm.txt b/requirements-rocm.txt index a846f929ef..7b42ee0673 100644 --- a/requirements-rocm.txt +++ b/requirements-rocm.txt @@ -2,7 +2,7 @@ ninja # For faster builds. typing-extensions>=4.8.0 starlette psutil -ray >= 2.5.1 +ray >= 2.9 sentencepiece # Required for LLaMA tokenizer. numpy tokenizers>=0.15.0 diff --git a/requirements.txt b/requirements.txt index 19871bdcc9..2bf527ccc3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ ninja # For faster builds. psutil -ray >= 2.5.1 +ray >= 2.9 sentencepiece # Required for LLaMA tokenizer. numpy torch == 2.1.2