python text_encoder_opc.py \
  --pretrained_model_name_or_path="/workspace/unlearncanvas_ckpt/diffusion/diffusion/diffuser/style50" \
  --dataset_name="unlearncanvas" --theme="Cats" \
  --use_ema  --resolution=512 --center_crop --random_flip \
  --train_batch_size=64 \
  --dataloader_num_worker=4 \
  --max_train_steps=16000 \
  --learning_rate=1e-05 \
  --lr_scheduler="constant" \
  --image_column="image" --caption_column="text" --report_to='wandb' \
  --output_dir="/workspace/sd_opc/text_encoder_only/unlearncanvas/Cats" \
  --tracker_project_name="text_encoder_only_unlearncanvas"