clones/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2024-09-22 11:07:28 +00:00
Code Issues Packages Projects Releases Wiki Activity
cc2dc12f6c
immich/machine-learning/pyproject.toml

71 lines
1.5 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.91.4
2023-12-19 03:34:19 +00:00
version = "1.91.4"
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]
feat(server,ml): remove image tagging (#5903) * remove image tagging * updated lock * fixed tests, improved logging * be nice * fixed tests
2023-12-21 01:47:56 +00:00
python = "=3.11.*"
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
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
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"
chore(ml): memory optimisations (#3934)
2023-08-31 23:30:53 +00:00
gunicorn = "^21.1.0"
feat(server,ml): remove image tagging (#5903) * remove image tagging * updated lock * fixed tests, improved logging * be nice * fixed tests
2023-12-21 01:47:56 +00:00
huggingface-hub = "^0.20.1"
tokenizers = "^0.15.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
[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 New Issue Copy Permalink