set INSTANCE_NAME="dbperson"
set CLASS_NAME="person"

set INSTANCE_DIR="./in/person"
set OUTPUT_DIR="./out/person"
set CLASS_DIR="./class/sd15_person"

set NUM_INSTANCE_IMAGES=20
set /a NUM_CLASS_IMAGES=%NUM_INSTANCE_IMAGES%*12
set /a MAX_NUM_STEPS = %NUM_INSTANCE_IMAGES%*80
set LR_SCHEDULE="polynomial"
set /a LR_WARMUP_STEPS=%MAX_NUM_STEPS%/10

set INSTANCE_PROMPT="%INSTANCE_NAME% %CLASS_NAME%"

call T:/programs/anaconda3/Scripts/activate.bat
call conda activate diffusers

accelerate launch train_dreambooth.py ^
  --pretrained_model_name_or_path="./models/sd15" ^
  --tokenizer_name=%TOKENIZER_NAME% ^
  --pretrained_vae_name_or_path="./vae" ^
  --instance_data_dir=%INSTANCE_DIR% ^
  --class_data_dir=%CLASS_DIR% ^
  --output_dir=%OUTPUT_DIR% ^
  --with_prior_preservation --prior_loss_weight=1.0 ^
  --instance_prompt=%INSTANCE_PROMPT% ^
  --class_prompt=%CLASS_NAME% ^
  --seed=1337 ^
  --resolution=512 ^
  --train_batch_size=1 ^
  --train_text_encoder ^
  --mixed_precision="fp16" ^
  --use_8bit_adam ^
  --gradient_accumulation_steps=1 ^
  --learning_rate=1e-6 ^
  --lr_scheduler=%LR_SCHEDULE% ^
  --lr_warmup_steps=%LR_WARMUP_STEPS% ^
  --num_class_images=%NUM_CLASS_IMAGES% ^
  --sample_batch_size=4 ^
  --max_train_steps=%MAX_NUM_STEPS% ^
  --not_cache_latents ^
  --save_interval=250
pause