diff --git a/Dockerfile b/Dockerfile index e00d649..08464be 100644 --- a/Dockerfile +++ b/Dockerfile @@ -102,7 +102,7 @@ RUN mkdir -p /wheels && \ RUN pip wheel \ pydantic orjson psutil pyzmq pynvml \ transformers==4.52.0 uvicorn fastapi IPython aiohttp \ - setproctitle uvloop sentencepiece triton pillow \ + setproctitle uvloop sentencepiece triton pillow cachetools \ -w /wheels ############################################################################### @@ -115,7 +115,7 @@ ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF- RUN apt-get update && apt-get install -y --no-install-recommends \ python3 python3-pip python3-distutils ca-certificates \ - libopenblas-dev libgomp1 libcupti-dev libnuma1 libopenmpi-dev openmpi-bin libnuma-dev libpng16-16 && \ + libopenblas-dev libgomp1 libcupti-dev libnuma1 libopenmpi-dev openmpi-bin libnuma-dev libpng16-16 libjpeg8 && \ rm -rf /var/lib/apt/lists/* && \ python3 -m pip install --no-cache-dir --upgrade pip