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
|
|
|
|
2023-03-23 07:33:45 +00:00
|
|
|
[tool.poetry.scripts]
|
|
|
|
serve = "api.main:serve"
|
2023-03-18 08:41:31 +00:00
|
|
|
|
|
|
|
[tool.poetry.dependencies]
|
|
|
|
python = "^3.10"
|
|
|
|
fastapi = "^0.94.1"
|
|
|
|
langchain = "^0.0.115"
|
|
|
|
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"
|
|
|
|
|
|
|
|
[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"
|
2023-03-23 11:23:40 +00:00
|
|
|
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"
|
2023-03-19 06:17:36 +00:00
|
|
|
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"
|