diff --git a/pyproject.toml b/pyproject.toml index 70c006a..cde247a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -10,8 +10,11 @@ readme = "README.md" python = "^3.10" pexpect = "^4.9.0" torch = "2.2.1" -#vllm = "0.4.1" +vllm = "0.4.1" openai = "1.30.5" +fire = "0.6.0" +wandb = "0.17.0" +tenacity = "8.3.0" [tool.poetry.build] generate-setup-file = false