Updated dockerfile to use docker image 2.4.0:1.18.0-500

huggingface · Oct 2, 2024 · 3329c46 · 3329c46
1 parent bab529c
commit 3329c46
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -42,7 +42,7 @@ COPY launcher launcher
 RUN cargo build --profile release-opt
 
 # Text Generation Inference base image
-FROM vault.habana.ai/gaudi-docker/1.17.0/ubuntu22.04/habanalabs/pytorch-installer-2.3.1:latest as base
+FROM artifactory-kfs.habana-labs.com/docker-local/1.18.0/ubuntu22.04/habanalabs/pytorch-installer-2.4.0:1.18.0-500 as base
 
 # Text Generation Inference base env
 ENV HF_HOME=/data \
@@ -97,5 +97,5 @@ FROM base
 COPY ./tgi-entrypoint.sh /tgi-entrypoint.sh
 RUN chmod +x /tgi-entrypoint.sh
 
-#ENTRYPOINT ["/tgi-entrypoint.sh"]
+ENTRYPOINT ["/tgi-entrypoint.sh"]
 # CMD ["--json-output"]