train-diffuser

Paused

nroggendorff commited on Sep 16

Commit

eeba825

verified ·

1 Parent(s): 76cadd1

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -7,7 +7,7 @@ ENV HOME=/home/user \
   HF_DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
   TRITON_CACHE_DIR=$HOME/.cache/.triton/
-ENV INIT=3 \
   INSTRUCT=false \
-  BATCH_SIZE=64 \
-  # FACTOR=24576

   HF_DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
   TRITON_CACHE_DIR=$HOME/.cache/.triton/
+ENV INIT=0 \
   INSTRUCT=false \
+  BATCH_SIZE=32 \
+  FACTOR=24576