Update Dockerfile

Dockerfile

@@ -27,8 +27,8 @@ RUN pip install --no-cache-dir --upgrade pip && \
 
 RUN mkdir /.cache && chmod 777 /.cache
 
-ARG OUTPUT_DIR=zelda-style-diffusion
-ARG INPUT_DS=nroggendorff/zelda-style
+ARG OUTPUT_DIR=fry-diffusion
+ARG INPUT_DS=nroggendorff/fries
 ARG BASE=glides/illustriousxl
 
 ENV OUTPUT_DIR=${OUTPUT_DIR} \
@@ -44,7 +44,7 @@ CMD bash -c "accelerate launch train.py \
   --gradient_accumulation_steps=1 \
   --learning_rate=3e-5 \
   --gradient_checkpointing \
-  --max_train_steps=3000 \
+  --max_train_steps=2000 \
   --lr_scheduler=cosine_with_restarts \
   --lr_warmup_steps=200 \
   --center_crop \