diff --git a/pyproject.toml b/pyproject.toml index a16fd81..ee6e0e0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -45,7 +45,7 @@ dependencies = [ "tensordict<0.6", "torchdata", "transformers", - "vllm<=0.6.3", + "vllm<=0.8.5", 'wandb', ] diff --git a/requirements.txt b/requirements.txt index 8c35b11..e7632ed 100644 --- a/requirements.txt +++ b/requirements.txt @@ -16,5 +16,5 @@ ray[data,train,tune,serve] tensordict<0.6 torchdata transformers -vllm<=0.6.3 +vllm<=0.8.5 wandb