diff --git a/Dockerfile b/Dockerfile index 59a6177..40d8f0a 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 cachetools msgspec blake3 cloudpickle \ + setproctitle uvloop sentencepiece triton pillow cachetools msgspec blake3 cloudpickle compressed-tensor \ -w /wheels ###############################################################################