Hugging Face's logo

Spaces:
nroggendorff
/
train-diffuser
Paused

App Files Community
nroggendorff commited on
Commit
76cadd1
·
verified ·
1 Parent(s): 735aa11

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -7,4 +7,7 @@ ENV HOME=/home/user \
7
  HF_DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
8
  TRITON_CACHE_DIR=$HOME/.cache/.triton/
9
 
10
- ENV INIT=3
 
 
 
 
7
  HF_DATASETS_CACHE=$HOME/.cache/huggingface/datasets \
8
  TRITON_CACHE_DIR=$HOME/.cache/.triton/
9
 
10
+ ENV INIT=3 \
11
+ INSTRUCT=false \
12
+ BATCH_SIZE=64 \
13
+ # FACTOR=24576