From 0b5859213e2858da16f02937644074d36ee3c1f8 Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-172-31-41-161.ec2.internal>
Date: Fri, 9 Jun 2023 16:25:14 +0000
Subject: [PATCH] Fixing the dockerfile (require triton + gcc for compiling).

---
 Dockerfile | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 576dab8d..3a146740 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -159,11 +159,14 @@ COPY --from=builder /usr/src/target/release/text-generation-router /usr/local/bi
 # Install launcher
 COPY --from=builder /usr/src/target/release/text-generation-launcher /usr/local/bin/text-generation-launcher
 
+RUN apt update && apt install build-essential g++ -y
+
 # AWS Sagemaker compatbile image
 FROM base as sagemaker
 
 COPY sagemaker-entrypoint.sh entrypoint.sh
 RUN chmod +x entrypoint.sh
+RUN pip install triton
 
 ENTRYPOINT ["./entrypoint.sh"]