From 0604c5cb83e4bab22d1d6b7ed47c0fedff2aad20 Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Mon, 8 Apr 2024 14:56:51 +0200
Subject: [PATCH] add py-cpuinfo

---
 Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index e8c3096e..360a6d2c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -210,8 +210,8 @@ COPY --from=vllm-builder /usr/src/vllm/build/lib.linux-x86_64-cpython-310 /opt/c
 COPY --from=mamba-builder /usr/src/mamba/build/lib.linux-x86_64-cpython-310/ /opt/conda/lib/python3.10/site-packages
 COPY --from=mamba-builder /usr/src/causal-conv1d/build/lib.linux-x86_64-cpython-310/ /opt/conda/lib/python3.10/site-packages
 
-# Install flash-attention dependencies
-RUN pip install einops prometheus_client --no-cache-dir
+# Install vllm/flash-attention dependencies
+RUN pip install einops py-cpuinfo prometheus_client --no-cache-dir
 
 # Install server
 COPY proto proto