Fix typo in vllm extra requirements

dottxt-ai · Dec 8, 2024 · dccaace · dccaace
1 parent c8e69c6
commit dccaace
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"]