train-diffuser

Paused

nroggendorff commited on Sep 19, 2025

Commit

5611e6b

verified ·

1 Parent(s): b81b6b9

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -8,5 +8,5 @@ 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=expandable_segments:True \
   FLASH_ATTENTION=1