diff --git a/pyproject.toml b/pyproject.toml index 4c4ffc8b8..f75afbae9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -136,7 +136,7 @@ cpu = [ "transformers>=5.6,<6", "triton>=2.0.0; sys_platform=='linux'", "trl>=0.23.0", - "vllm==0.20.0; sys_platform=='linux'", + "vllm==0.22.0; sys_platform=='linux'", ] cu129 = [ @@ -160,7 +160,7 @@ cu129 = [ "transformers>=5.6,<6", "triton>=2.0.0; sys_platform == 'linux'", "trl>=0.23.0", - "vllm==0.20.0+cu129; sys_platform == 'linux'", + "vllm==0.22.0; sys_platform == 'linux'", ] # at some point, do per-subpackage dependencies