diff --git a/Dockerfile b/Dockerfile index 269710e..29ebea2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -103,7 +103,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 cachetools msgspec blake3 cloudpickle compressed-tensors einops openai \ + setproctitle uvloop sentencepiece triton pillow cachetools msgspec blake3 cloudpickle compressed-tensors einops openai py-cpuinfo \ -w /wheels ###############################################################################