diff --git a/elise/pyproject.toml b/elise/pyproject.toml index deb5e7f286ca65063a7e492c095ad9df9a33866d..a515fce98bee5989d610d8f5fee1ec289bca248f 100644 --- a/elise/pyproject.toml +++ b/elise/pyproject.toml @@ -7,17 +7,23 @@ readme = "README.md" [tool.poetry.scripts] benchmark = "elise.benchmark:main" +annotate = "elise.mwoz_annotation:main" +user_scripts = "elise.user_scripts:main" +diagen = "elise.multi_agent_generation:run_dialogue" [tool.poetry.dependencies] -python = "^3.9" +python = "^3.10" torch = "^2.1.2" -vllm = {git = "https://github.com/vllm-project/vllm.git", rev = "main"} -outlines = "^0.0.24" +vllm = "^3.1" +outlines = "^0.0.33" ray = "^2.9.1" huggingface = "^0.0.1" polars = {extras = ["numpy", "plot"], version = "^0.20.5"} +accelerate = "^0.26.1" +auto-gptq = "^0.6.0" +optimum = "^1.16.2" [build-system] requires = ["poetry-core"] -build-backend = "poetry.core.masonry.api" +build-backend = "poetry.core.masonry.api" \ No newline at end of file