From dccaaced6c9972f1c75dab0de83afbd68cfde933 Mon Sep 17 00:00:00 2001 From: Hicham Randrianarivo Date: Tue, 3 Dec 2024 10:49:34 +0100 Subject: [PATCH] Fix typo in vllm extra requirements --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index f738c965f..2001f4c40 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -45,7 +45,7 @@ dependencies = [ dynamic = ["version"] [project.optional-dependencies] -vllm = ["vllm", "transformers", "numpy2"] +vllm = ["vllm", "transformers", "numpy<2"] transformers = ["transformers", "accelerate", "datasets", "numpy<2"] mlxlm = ["mlx-lm", "datasets"] openai = ["openai"]