pyproject.toml

[tool.poetry]
name = "align-system"
version = "0.5.6"
description = ""
authors = ["David Joy <10147749+dmjoy@users.noreply.github.com>"]
readme = "README.md"
packages = [{include = "align_system"}]

[[tool.poetry.source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu118"
priority = "supplemental"

[tool.poetry.dependencies]
python = ">=3.9,<4.0"
torch = { version = "^2.0.1", source = "pytorch" }
transformers = ">=4.46.2"
llama-index = "^0.8.21"
langchain = "^0.0.308"
sentence-transformers = "^2.2.2"
accelerate = ">=0.26.0"
pypdf = "^3.15.5"
pycryptodome = "^3.18.0"
einops = "^0.6.1"
requests = "^2.31.0"
bert-score = "^0.3.13"
rich = "^13.6.0"
rouge-score = "^0.1.2"
swagger-client = {git = "https://github.com/NextCenturyCorporation/itm-evaluation-client.git", rev = "development"}
hydra-core = "^1.3.2"
outlines = "^0.0.46"
setuptools = "^70.1.1"
sentencepiece = "^0.2.0"
protobuf = "^5.28.3"

[tool.poetry.scripts]
run_align_system = 'align_system.cli.run_align_system:main'

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"