train-diffuser

Paused

nroggendorff commited on Sep 22, 2025

Commit

059540c

verified ·

1 Parent(s): 5611e6b

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -8,5 +8,6 @@ ENV DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
   TRITON_CACHE_DIR=$HOME/.cache/.triton/
 ENV TOKENIZERS_PARALLELISM=false \
-  PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True \
-  FLASH_ATTENTION=1

   TRITON_CACHE_DIR=$HOME/.cache/.triton/
 ENV TOKENIZERS_PARALLELISM=false \
+  PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:128 \
+  FLASH_ATTENTION=1 \
+  ACCELERATE_USE_DEEPSPEED=true