EVAL/pyproject.toml

56 lines
1.2 KiB
TOML
Raw Normal View History

2023-03-18 08:41:31 +00:00
[tool.poetry]
2023-03-20 08:27:20 +00:00
name = "eval"
2023-03-18 08:41:31 +00:00
version = "0.1.0"
description = ""
authors = ["Taeho Lee <taeho@corca.ai>", "Chung Hwan Han <hanch@corca.ai>"]
2023-03-23 08:53:25 +00:00
packages = [{include = "api"}]
2023-03-18 08:41:31 +00:00
[tool.poetry.scripts]
serve = "api.main:serve"
dev = "api.main:dev"
2023-03-18 08:41:31 +00:00
[tool.poetry.dependencies]
python = "^3.10"
fastapi = "^0.94.1"
langchain = "^0.0.139"
2023-03-18 08:41:31 +00:00
diffusers = "^0.14.0"
pydantic = "^1.10.6"
tenacity = "^8.2.2"
2023-04-03 12:40:58 +00:00
llama-index = "0.4.29"
2023-03-18 08:41:31 +00:00
python-dotenv = "^1.0.0"
pillow = "^9.4.0"
2023-03-18 09:51:31 +00:00
boto3 = "^1.26.94"
uvicorn = "^0.21.1"
2023-04-03 12:40:58 +00:00
python-ptrace = "^0.9.8"
jinja2 = "^3.1.2"
python-multipart = "^0.0.6"
celery = "^5.2.7"
redis = "^4.5.4"
2023-04-03 12:40:58 +00:00
[tool.poetry.group.gpu]
optional = true
[tool.poetry.group.gpu.dependencies]
torch = "^2.0.0"
2023-03-18 09:51:31 +00:00
accelerate = "^0.17.1"
transformers = {git = "https://github.com/huggingface/transformers.git", rev = "main"}
sentencepiece = "^0.1.97"
bitsandbytes = "^0.37.2"
2023-03-18 08:41:31 +00:00
[tool.poetry.group.tools]
optional = true
[tool.poetry.group.tools.dependencies]
psycopg2-binary = "^2.9.5"
wikipedia = "^1.4.0"
google-search-results = "^2.4.2"
beautifulsoup4 = "^4.11.2"
2023-03-18 08:41:31 +00:00
2023-03-18 12:26:19 +00:00
[tool.poetry.group.dev.dependencies]
black = "^23.1.0"
2023-03-18 08:41:31 +00:00
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"