diff --git a/pyproject.toml b/pyproject.toml index 3476a8b..5584191 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,7 +25,7 @@ dependencies = [ "torch==2.4.0", "tqdm==4.66.5", "transformers==4.45.2", - "vllm==0.5.5", + "vllm==0.7.0", "wandb==0.18.3" ]