diff --git a/pyproject.toml b/pyproject.toml index 69733f665..ee200f861 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -97,7 +97,7 @@ nanotron = [ "tensorboardX" ] tensorboardX = ["tensorboardX"] -vllm = ["vllm>=0.10.0", "ray", "more_itertools"] +vllm = ["vllm>=0.10.0,<0.10.2", "ray", "more_itertools"] quality = ["ruff>=v0.11.0","pre-commit"] tests = ["pytest>=7.4.0","deepdiff"] dev = ["lighteval[accelerate,quality,tests,multilingual,math,extended_tasks,vllm]"]