mirror of
https://github.com/immich-app/immich.git
synced 2024-11-28 09:33:27 +02:00
95cfe22866
* cuda and openvino ep, refactor, update dockerfile * updated workflow * typing fixes * added tests * updated ml test gh action * updated README * updated docker-compose * added compute to hwaccel.yml * updated gh matrix updated gh matrix updated gh matrix updated gh matrix updated gh matrix give up * remove cuda/arm64 build * add hwaccel image tags to docker-compose * remove unnecessary quotes * add suffix to git tag * fixed kwargs in base model * armnn ld_library_path * update pyproject.toml * add armnn workflow * formatting * consolidate hwaccel files, update docker compose * update hw transcoding docs * add ml hwaccel docs * update dev and prod docker-compose * added armnn prerequisite docs * support 3.10 * updated docker-compose comments * formatting * test coverage * don't set arena extend strategy for openvino * working openvino * formatting * fix dockerfile * added type annotation * add wsl configuration for openvino * updated lock file * copy python3 * comment out extends section * fix platforms * simplify workflow suffix tagging * simplify aio transcoding doc * update docs and workflow for `hwaccel.yml` change * revert docs
96 lines
2.0 KiB
TOML
96 lines
2.0 KiB
TOML
[tool.poetry]
|
|
name = "machine-learning"
|
|
version = "1.93.3"
|
|
description = ""
|
|
authors = ["Hau Tran <alex.tran1502@gmail.com>"]
|
|
readme = "README.md"
|
|
packages = [{include = "app"}]
|
|
|
|
[tool.poetry.dependencies]
|
|
python = ">=3.10,<3.12"
|
|
insightface = ">=0.7.3,<1.0"
|
|
opencv-python-headless = ">=4.7.0.72,<5.0"
|
|
pillow = ">=9.5.0,<11.0"
|
|
fastapi = ">=0.95.2,<1.0"
|
|
uvicorn = {extras = ["standard"], version = ">=0.22.0,<1.0"}
|
|
pydantic = "^1.10.8"
|
|
aiocache = ">=0.12.1,<1.0"
|
|
rich = ">=13.4.2"
|
|
ftfy = ">=6.1.1"
|
|
setuptools = "^68.0.0"
|
|
python-multipart = ">=0.0.6,<1.0"
|
|
orjson = ">=3.9.5"
|
|
gunicorn = ">=21.1.0"
|
|
huggingface-hub = ">=0.20.1,<1.0"
|
|
tokenizers = ">=0.15.0,<1.0"
|
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
mypy = ">=1.3.0"
|
|
black = ">=23.3.0"
|
|
pytest = ">=7.3.1"
|
|
locust = ">=2.15.1"
|
|
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.group.cpu]
|
|
optional = true
|
|
|
|
[tool.poetry.group.cpu.dependencies]
|
|
onnxruntime = "^1.15.0"
|
|
|
|
[tool.poetry.group.cuda]
|
|
optional = true
|
|
|
|
[tool.poetry.group.cuda.dependencies]
|
|
onnxruntime-gpu = "^1.15.0"
|
|
|
|
[tool.poetry.group.openvino]
|
|
optional = true
|
|
|
|
[tool.poetry.group.openvino.dependencies]
|
|
onnxruntime-openvino = ">=1.15.0,<1.16.0"
|
|
|
|
[tool.poetry.group.armnn]
|
|
optional = true
|
|
|
|
[tool.poetry.group.armnn.dependencies]
|
|
onnxruntime = "^1.15.0"
|
|
|
|
[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
|
|
ignore_missing_imports = true
|
|
|
|
[tool.pydantic-mypy]
|
|
init_forbid_extra = true
|
|
init_typed = true
|
|
warn_required_dynamic_aliases = true
|
|
warn_untyped_fields = true
|
|
|
|
[tool.ruff]
|
|
line-length = 120
|
|
target-version = "py311"
|
|
select = ["E", "F", "I"]
|
|
|
|
[tool.ruff.per-file-ignores]
|
|
"test_main.py" = ["F403"]
|
|
|
|
[tool.black]
|
|
line-length = 120
|
|
target-version = ['py311']
|
|
|
|
[tool.pytest.ini_options]
|
|
markers = ["providers"]
|