From bbe95ca9e9079c5c7cd29dff83fb759d1771c89b Mon Sep 17 00:00:00 2001
From: Funtowicz Morgan <mfuntowicz@users.noreply.github.com>
Date: Thu, 28 Nov 2024 09:53:15 +0100
Subject: [PATCH] Update Dockerfile.llamacpp as per review

Co-authored-by: Hugo Larcher <hugo.larcher@huggingface.co>
---
 Dockerfile.llamacpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Dockerfile.llamacpp b/Dockerfile.llamacpp
index 916f885a..44583b09 100644
--- a/Dockerfile.llamacpp
+++ b/Dockerfile.llamacpp
@@ -64,7 +64,9 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     apt install -y \
     numactl \
     openssl \
-    python3.11-dev
+    python3.11-dev \
+    python3.11-venv \
+    ibgomp1
 
 COPY --from=builder /usr/src/text-generation-inference/target/release-opt/text-generation-backend-llamacpp /usr/src/text-generation-inference/text-generation-launcher
 COPY --from=builder /usr/src/text-generation-inference/dist /usr/