comeback vllm

turbo-llm · Sep 28, 2024 · f254065 · f254065
1 parent 9ce44c6
commit f254065
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -70,7 +70,7 @@ annotated-types = "^0.6.0"
 
 [tool.poetry.extras]
 deepspeed = ["deepspeed", "accelerate"]
-gpu = ["faiss-gpu", "xformers"]
+gpu = ["faiss-gpu", "xformers", "vllm"]
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]