diff --git a/pyproject.toml b/pyproject.toml index cb9d543..0b92c37 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -39,7 +39,7 @@ dependencies = [ "tiktoken==0.9.0", "transformers==5.0.0rc3", "trl==0.17.0", - "vllm==0.23.0", + "vllm==0.24.0", ] [project.urls] diff --git a/requirements.txt b/requirements.txt index 302e3cc..5a5baf9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,4 +15,4 @@ submitit tiktoken==0.9.0 transformers==5.0.0rc3 trl==0.17.0 -vllm==0.23.0 +vllm==0.24.0