Merge branch #nroggendorff/train-llama' into 'nroggendorff/train-diffuser'

Dockerfile

@@ -2,13 +2,15 @@ FROM nroggendorff/train-diffuser:latest
 
 ENV HOME=/home/user
 ENV HF_HOME=$HOME/.cache/huggingface \
-    PATH=$HOME/.local/bin:$PATH \
-    DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
-    HF_DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
-    TRITON_CACHE_DIR=$HOME/.cache/.triton/
+  PATH=$HOME/.local/bin:$PATH \
+  DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
+  HF_DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
+  TRITON_CACHE_DIR=$HOME/.cache/.triton/
 
-ENV PYTHONUNBUFFERED=1
+ENV PYTHONUNBUFFERED=1 \
+  TQDM_POSITION=-1 \
+  TQDM_MININTERVAL=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
+  echo "export STARTUP_DURATION_TIMEOUT=$STARTUP_TIMEOUT" >> ~/.bashrc