Add PYTHONBUFFERED=1 to Dockerfile.{cpu,gpu}

alvarobartt · Mar 6, 2024 · 0e0cf1a · 0e0cf1a
1 parent 6095bbd
commit 0e0cf1a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/vertex_ai_huggingface_inference_toolkit/dockerfiles/Dockerfile.cpu b/src/vertex_ai_huggingface_inference_toolkit/dockerfiles/Dockerfile.cpu
@@ -4,6 +4,7 @@ LABEL maintainer="Alvaro Bartolome"
 
 ARG DEBIAN_FRONTEND=noninteractive
 ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONBUFFERED=1
 
 RUN mkdir -m 777 -p /usr/app /home
 WORKDIR /usr/app
@@ -30,4 +31,3 @@ ENV PREDICTOR_CLASS=TransformersPredictor
 
 EXPOSE 8080
 ENTRYPOINT ["python", "-m", "google.cloud.aiplatform.prediction.model_server"]
-
diff --git a/src/vertex_ai_huggingface_inference_toolkit/dockerfiles/Dockerfile.gpu b/src/vertex_ai_huggingface_inference_toolkit/dockerfiles/Dockerfile.gpu
@@ -4,6 +4,7 @@ LABEL maintainer="Alvaro Bartolome"
 
 ARG DEBIAN_FRONTEND=noninteractive
 ENV PYTHONDONTWRITEBYTECODE=1
+ENV PYTHONBUFFERED=1
 
 RUN mkdir -m 777 -p /usr/app /home
 WORKDIR /usr/app