beenull/immich
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity
immich/machine-learning/pyproject.toml

81 lines
1.8 KiB
TOML
Raw Normal View History

chore(ml): updated dockerfile, added typing, packaging (#2642) * updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose
2023-06-05 14:40:48 +00:00
[tool.poetry]
name = "machine-learning"
Version v1.85.0
2023-11-08 03:20:59 +00:00
version = "1.85.0"
chore(ml): updated dockerfile, added typing, packaging (#2642) * updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose
2023-06-05 14:40:48 +00:00
description = ""
authors = ["Hau Tran <alex.tran1502@gmail.com>"]
readme = "README.md"
packages = [{include = "app"}]
[tool.poetry.dependencies]
python = "^3.11"
torch = [
feat(ml): export clip models to ONNX and host models on Hugging Face (#4700) * export clip models * export to hf refactored export code * export mclip, general refactoring cleanup * updated conda deps * do transforms with pillow and numpy, add tokenization config to export, general refactoring * moved conda dockerfile, re-added poetry * minor fixes * updated link * updated tests * removed `requirements.txt` from workflow * fixed mimalloc path * removed torchvision * cleaner np typing * review suggestions * update default model name * update test
2023-10-31 10:02:04 +00:00
{markers = "platform_machine == 'arm64' or platform_machine == 'aarch64'", version = "=2.1.0", source = "pypi"},
{markers = "platform_machine == 'amd64' or platform_machine == 'x86_64'", version = "=2.1.0", source = "pytorch-cpu"}
chore(ml): updated dockerfile, added typing, packaging (#2642) * updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose
2023-06-05 14:40:48 +00:00
]
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"
feat(ml): model unloading (#2661) * model cache * fixed revalidation when using cache namespace * fixed ttl not being set, added lock
2023-06-07 01:48:51 +00:00
aiocache = "^0.12.1"
feat(ml)!: switch image classification and CLIP models to ONNX (#3809)
2023-08-25 04:28:51 +00:00
optimum = "^1.9.1"
rich = "^13.4.2"
ftfy = "^6.1.1"
setuptools = "^68.0.0"
feat(ml)!: customizable ML settings (#3891) * consolidated endpoints, added live configuration * added ml settings to server * added settings dashboard * updated deps, fixed typos * simplified modelconfig updated tests * Added ml setting accordion for admin page updated tests * merge `clipText` and `clipVision` * added face distance setting clarified setting * add clip mode in request, dropdown for face models * polished ml settings updated descriptions * update clip field on error * removed unused import * add description for image classification threshold * pin safetensors for arm wheel updated poetry lock * moved dto * set model type only in ml repository * revert form-data package install use fetch instead of axios * added slotted description with link updated facial recognition description clarified effect of disabling tasks * validation before model load * removed unnecessary getconfig call * added migration * updated api updated api updated api --------- Co-authored-by: Alex Tran <alex.tran1502@gmail.com>
2023-08-29 13:58:00 +00:00
python-multipart = "^0.0.6"
orjson = "^3.9.5"
safetensors = "0.3.2"
chore(ml): memory optimisations (#3934)
2023-08-31 23:30:53 +00:00
gunicorn = "^21.1.0"
chore(ml): updated dockerfile, added typing, packaging (#2642) * updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose
2023-06-05 14:40:48 +00:00
[tool.poetry.group.dev.dependencies]
mypy = "^1.3.0"
black = "^23.3.0"
pytest = "^7.3.1"
added locustfile (#2926)
2023-06-25 18:20:45 +00:00
locust = "^2.15.1"
chore(ml): added testing and github workflow (#2969) * added testing * github action for python, made mypy happy * formatted with black * minor fixes and styling * test model cache * cache test dependencies * narrowed model cache tests * moved endpoint tests to their own class * cleaned up fixtures * formatting * removed unused dep
2023-06-27 23:21:33 +00:00
httpx = "^0.24.1"
pytest-asyncio = "^0.21.0"
moved deps (#3379)
2023-07-22 20:04:52 +00:00
pytest-cov = "^4.1.0"
ruff = "^0.0.272"
refactor(ml): model downloading (#3545) * download facial recognition models * download hf models * simplified logic * updated `predict` for facial recognition * ensure download method is called * fixed repo_id for clip * fixed download destination * use st's own `snapshot_download` * conditional download * fixed predict method * check if loaded * minor fixes * updated mypy overrides * added pytest-mock * updated tests * updated lock
2023-08-06 02:45:13 +00:00
pytest-mock = "^3.11.1"
chore(ml): updated dockerfile, added typing, packaging (#2642) * updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose
2023-06-05 14:40:48 +00:00
[[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
feat(ml): export clip models to ONNX and host models on Hugging Face (#4700) * export clip models * export to hf refactored export code * export mclip, general refactoring cleanup * updated conda deps * do transforms with pillow and numpy, add tokenization config to export, general refactoring * moved conda dockerfile, re-added poetry * minor fixes * updated link * updated tests * removed `requirements.txt` from workflow * fixed mimalloc path * removed torchvision * cleaner np typing * review suggestions * update default model name * update test
2023-10-31 10:02:04 +00:00
ignore_missing_imports = true
chore(ml): updated dockerfile, added typing, packaging (#2642) * updated dockerfile, added typing, packaging apply env change * added arm64 support * added ml version pump, second try for arm64 * added linting config to pyproject.toml * renamed ml input field * fixed linter config * fixed dev docker compose
2023-06-05 14:40:48 +00:00
[tool.pydantic-mypy]
init_forbid_extra = true
init_typed = true
warn_required_dynamic_aliases = true
chore(ml): added testing and github workflow (#2969) * added testing * github action for python, made mypy happy * formatted with black * minor fixes and styling * test model cache * cache test dependencies * narrowed model cache tests * moved endpoint tests to their own class * cleaned up fixtures * formatting * removed unused dep
2023-06-27 23:21:33 +00:00
warn_untyped_fields = 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']
Reference in a new issue Copy permalink