diff --git a/machine-learning/Dockerfile b/machine-learning/Dockerfile index 01b083815..7b9df7b38 100644 --- a/machine-learning/Dockerfile +++ b/machine-learning/Dockerfile @@ -1,4 +1,4 @@ -FROM python:3.11-bookworm@sha256:ba7a7ac30c38e119c4304f98ef0e188f90f4f67a958bb6899da9defb99bfb471 as builder +FROM python:3.12-bookworm@sha256:5eba34eb667213abb09a4c470365180d5706076f76945e49b963ac15d428a684 as builder ENV PYTHONDONTWRITEBYTECODE=1 \ PYTHONUNBUFFERED=1 \ @@ -13,7 +13,7 @@ ENV VIRTUAL_ENV="/opt/venv" PATH="/opt/venv/bin:${PATH}" COPY poetry.lock pyproject.toml ./ RUN poetry install --sync --no-interaction --no-ansi --no-root --only main -FROM python:3.11-slim-bookworm@sha256:cc758519481092eb5a4a5ab0c1b303e288880d59afc601958d19e95b300bc86b +FROM python:3.12-slim-bookworm@sha256:19a6235339a74eca01227b03629f63b6f5020abc21142436eced6ec3a9839a76 RUN apt-get update && apt-get install -y --no-install-recommends tini libmimalloc2.0 && rm -rf /var/lib/apt/lists/* diff --git a/machine-learning/pyproject.toml b/machine-learning/pyproject.toml index b7c831429..a0622ae9e 100644 --- a/machine-learning/pyproject.toml +++ b/machine-learning/pyproject.toml @@ -7,7 +7,7 @@ readme = "README.md" packages = [{include = "app"}] [tool.poetry.dependencies] -python = "~3.11" +python = "~3.12.0" torch = [ {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"} @@ -17,8 +17,8 @@ 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"} +fastapi = "^0.104.0" +uvicorn = {extras = ["standard"], version = "^0.24.0"} pydantic = "^1.10.8" aiocache = "^0.12.1" optimum = "^1.9.1" @@ -27,7 +27,7 @@ ftfy = "^6.1.1" setuptools = "^68.0.0" python-multipart = "^0.0.6" orjson = "^3.9.5" -safetensors = "0.3.2" +safetensors = "0.4.1" gunicorn = "^21.1.0" [tool.poetry.group.dev.dependencies] @@ -35,10 +35,10 @@ mypy = "^1.3.0" black = "^23.3.0" pytest = "^7.3.1" locust = "^2.15.1" -httpx = "^0.24.1" +httpx = "^0.25.0" pytest-asyncio = "^0.21.0" pytest-cov = "^4.1.0" -ruff = "^0.0.272" +ruff = "^0.1.0" pytest-mock = "^3.11.1" [[tool.poetry.source]]