Update Dockerfile

Dockerfile

@@ -7,6 +7,8 @@ ENV HF_HOME=$HOME/.cache/huggingface \
     HF_DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
     TRITON_CACHE_DIR=$HOME/.cache/.triton/
 
+ENV PYTHONUNBUFFERED=1
+
 COPY README.md /tmp/README.md
 RUN STARTUP_TIMEOUT=$(awk '/^startup_duration_timeout:/ {print $2}' /tmp/README.md) && \
     echo "export STARTUP_DURATION_TIMEOUT=$STARTUP_TIMEOUT" >> ~/.bashrc