add dependency

b570fce7 · baberabb · af884028 · b570fce7
Commit b570fce7 authored Nov 23, 2023 by baberabb
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

pyproject.toml pyproject.toml +3 -1

No files found.
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -71,6 +71,7 @@ promptsource = [
 gptq = ["auto-gptq[triton] @ git+https://github.com/PanQiWei/AutoGPTQ"]
 anthropic = ["anthropic"]
 openai = ["openai", "tiktoken"]
+vllm = ["vllm"]
 all = [
    "lm_eval[dev]",
    "lm_eval[testing]",
@@ -80,5 +81,6 @@ all = [
    "lm_eval[promptsource]",
    "lm_eval[gptq]",
    "lm_eval[anthropic]",
-    "lm_eval[openai]"
+    "lm_eval[openai]",
+    "lm_eval[vllm]",
 ]