diff --git a/elise/pyproject.toml b/elise/pyproject.toml index 5eb66fa0511771f3bf14710d6000b56e1b9bc1b7..be97f38da7699970e9743282985dd5eb2f19fd78 100644 --- a/elise/pyproject.toml +++ b/elise/pyproject.toml @@ -14,8 +14,8 @@ diagen = "elise.multi_agent_generation:run_dialogue" [tool.poetry.dependencies] python = "^3.9" torch = "^2.1.2" -vllm = "^0.3.2" -outlines = "^0.0.33" +vllm = {git = "https://github.com/vllm-project/vllm.git", rev = "main"} +outlines = "^0.0.24" ray = "^2.9.1" huggingface = "^0.0.1" polars = {extras = ["numpy", "plot"], version = "^0.20.5"}