dreambooth_run

accelerate launch examples/dreambooth/train_dreambooth_lora_flux.py \
  --pretrained_model_name_or_path=black-forest-labs/FLUX.1-dev \
  --dataset_name="google/dreambooth" \
  --output_dir="./big" \
  --mixed_precision="no" \
  --instance_prompt="a photo of wac linear pendant, black and white, white background" \
  --caption_column="prompt" \
  --resolution=512 \
  --train_batch_size=1 \
  --gradient_accumulation_steps=2 \
  --learning_rate=5e-5 \
  --report_to="wandb" \
  --lr_scheduler="constant" \
  --lr_warmup_steps=0 \
  --max_train_steps=2000 \
  --repeats=2 \
  --validation_prompt="a photo of wac linear pendant, black and white, white background" \
  --validation_epochs=50 \
  --num_validation_images=1 \
  --rank=16 \
  --checkpointing_steps=200 \
  --seed="0" \
  --gradient_accumulation=2 \
  --gradient_checkpointing \
  --use_8bit_adam \
  --cache_latents