Update Dockerfile

Dockerfile

@@ -12,4 +12,4 @@ RUN mkdir /app/zelda-xl && chmod 777 /app/zelda-xl
 
 RUN accelerate config default
 
-CMD ["accelerate", "launch", "train.py", "--mixed_precision=fp16", "--learning_rate=1e-5", "--gradient_checkpointing", "--max_train_steps=360", "--lr_scheduler=constant", "--lr_warmup_steps=0", "--use_ema", "--center_crop", "--random_flip", "--pretrained_model_name_or_path=nroggendorff/animexl", "--dataset_name=nroggendorff/zelda", "--resolution=1024", "--train_batch_size=4", "--output_dir=zelda-xl", "--push_to_hub"]
+CMD ["accelerate", "launch", "train.py", "--mixed_precision=fp16", "--learning_rate=1e-5", "--gradient_checkpointing", "--max_train_steps=360", "--lr_scheduler=constant", "--lr_warmup_steps=0", "--use_ema", "--center_crop", "--random_flip", "--pretrained_model_name_or_path=nroggendorff/animexl", "--dataset_name=nroggendorff/zelda", "--resolution=1024", "--train_batch_size=1", "--output_dir=zelda-xl", "--push_to_hub"]