Update Dockerfile

Dockerfile

@@ -1,21 +1,12 @@
 FROM nroggendorff/train-llama:latest
 
-ENV HOME=/home/user \
-    HF_HOME=$HOME/.cache/huggingface \
-    PATH=$HOME/.local/bin:$PATH
-ENV DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
+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/
 
-ENV CUDA_LAUNCH_BLOCKING=1 \
-    NCCL_DEBUG=WARN \
-    NCCL_ASYNC_ERROR_HANDLING=1 \
-    OMP_NUM_THREADS=1 \
-    TOKENIZERS_PARALLELISM=false \
-    PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128 \
-    FLASH_ATTENTION=1 \
-    ACCELERATE_USE_DEEPSPEED=true
-
 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