diff --git a/Dockerfile b/Dockerfile
index ea72ed7..8cc988a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -105,7 +105,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 py-cpuinfo dill partial_json_parser python-multipart \
+    setproctitle uvloop sentencepiece triton pillow cachetools msgspec blake3 cloudpickle compressed-tensors einops openai py-cpuinfo dill partial_json_parser python-multipart torchao \
     -w /wheels
 
 ###############################################################################