107 lines
2.6 KiB
TOML
107 lines
2.6 KiB
TOML
[tool.poetry]
|
|
name = "machine-learning"
|
|
version = "1.76.0"
|
|
description = ""
|
|
authors = ["Hau Tran <alex.tran1502@gmail.com>"]
|
|
readme = "README.md"
|
|
packages = [{include = "app"}]
|
|
|
|
[tool.poetry.dependencies]
|
|
python = "^3.11"
|
|
torch = [
|
|
{markers = "platform_machine == 'arm64' or platform_machine == 'aarch64'", version = "=2.0.1", source = "pypi"},
|
|
{markers = "platform_machine == 'amd64' or platform_machine == 'x86_64'", version = "=2.0.1", source = "pytorch-cpu"}
|
|
]
|
|
transformers = "^4.29.2"
|
|
onnxruntime = "^1.15.0"
|
|
insightface = "^0.7.3"
|
|
opencv-python-headless = "^4.7.0.72"
|
|
pillow = "^9.5.0"
|
|
fastapi = "^0.95.2"
|
|
uvicorn = {extras = ["standard"], version = "^0.22.0"}
|
|
pydantic = "^1.10.8"
|
|
aiocache = "^0.12.1"
|
|
optimum = "^1.9.1"
|
|
torchvision = [
|
|
{markers = "platform_machine == 'arm64' or platform_machine == 'aarch64'", version = "=0.15.2", source = "pypi"},
|
|
{markers = "platform_machine == 'amd64' or platform_machine == 'x86_64'", version = "=0.15.2", source = "pytorch-cpu"}
|
|
]
|
|
rich = "^13.4.2"
|
|
ftfy = "^6.1.1"
|
|
setuptools = "^68.0.0"
|
|
open-clip-torch = "^2.20.0"
|
|
python-multipart = "^0.0.6"
|
|
orjson = "^3.9.5"
|
|
safetensors = "0.3.2"
|
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
mypy = "^1.3.0"
|
|
black = "^23.3.0"
|
|
pytest = "^7.3.1"
|
|
locust = "^2.15.1"
|
|
gunicorn = "^20.1.0"
|
|
httpx = "^0.24.1"
|
|
pytest-asyncio = "^0.21.0"
|
|
pytest-cov = "^4.1.0"
|
|
ruff = "^0.0.272"
|
|
pytest-mock = "^3.11.1"
|
|
|
|
[[tool.poetry.source]]
|
|
name = "pytorch-cpu"
|
|
url = "https://download.pytorch.org/whl/cpu"
|
|
priority = "explicit"
|
|
|
|
[build-system]
|
|
requires = ["poetry-core"]
|
|
build-backend = "poetry.core.masonry.api"
|
|
|
|
[tool.mypy]
|
|
python_version = "3.11"
|
|
plugins = "pydantic.mypy"
|
|
follow_imports = "silent"
|
|
warn_redundant_casts = true
|
|
disallow_any_generics = true
|
|
check_untyped_defs = true
|
|
disallow_untyped_defs = true
|
|
|
|
[tool.pydantic-mypy]
|
|
init_forbid_extra = true
|
|
init_typed = true
|
|
warn_required_dynamic_aliases = true
|
|
warn_untyped_fields = true
|
|
|
|
[[tool.mypy.overrides]]
|
|
module = [
|
|
"huggingface_hub",
|
|
"transformers",
|
|
"cv2",
|
|
"insightface.model_zoo",
|
|
"insightface.utils.face_align",
|
|
"insightface.utils.storage",
|
|
"onnxruntime",
|
|
"optimum",
|
|
"optimum.pipelines",
|
|
"optimum.onnxruntime",
|
|
"clip_server.model.clip",
|
|
"clip_server.model.clip_onnx",
|
|
"clip_server.model.pretrained_models",
|
|
"clip_server.model.tokenization",
|
|
"torchvision.transforms",
|
|
"aiocache.backends.memory",
|
|
"aiocache.lock",
|
|
"aiocache.plugins"
|
|
]
|
|
ignore_missing_imports = true
|
|
|
|
[tool.ruff]
|
|
line-length = 120
|
|
target-version = "py311"
|
|
select = ["E", "F", "I"]
|
|
ignore = ["F401"]
|
|
|
|
[tool.ruff.per-file-ignores]
|
|
"test_main.py" = ["F403"]
|
|
|
|
[tool.black]
|
|
line-length = 120
|
|
target-version = ['py311']
|