History

Suraj Patil c228331068 [examples] add check_min_version (#1550 ) * add check_min_version for examples * move __version__ to the top * Apply suggestions from code review Co-authored-by: Pedro Cuenca <pedro@huggingface.co> * fix comment * fix error_message * adapt the install message Co-authored-by: Pedro Cuenca <pedro@huggingface.co>		2022-12-06 14:36:50 +01:00
..
README.md	Research folder (#1553 )	2022-12-05 17:58:35 +01:00
requirements.txt	Research folder (#1553 )	2022-12-05 17:58:35 +01:00
train_dreambooth_inpaint.py	[examples] add check_min_version (#1550 )	2022-12-06 14:36:50 +01:00

README.md

Dreambooth for the inpainting model

This script was added by @thedarkzeno .

Please note that this script is not actively maintained, you can open an issue and tag @thedarkzeno or @patil-suraj though.

export MODEL_NAME="runwayml/stable-diffusion-inpainting"
export INSTANCE_DIR="path-to-instance-images"
export OUTPUT_DIR="path-to-save-model"

accelerate launch train_dreambooth_inpaint.py \
  --pretrained_model_name_or_path=$MODEL_NAME  \
  --instance_data_dir=$INSTANCE_DIR \
  --output_dir=$OUTPUT_DIR \
  --instance_prompt="a photo of sks dog" \
  --resolution=512 \
  --train_batch_size=1 \
  --gradient_accumulation_steps=1 \
  --learning_rate=5e-6 \
  --lr_scheduler="constant" \
  --lr_warmup_steps=0 \
  --max_train_steps=400

The script is also compatible with prior preservation loss and gradient checkpointing