File size: 708 Bytes
ece766c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
export MODEL_NAME="stable-diffusion/stable-diffusion-1-5"
export INSTANCE_DIR="data/training/"
export OUTPUT_DIR="output/lora/"
export CLASS_DIR="data/lora_class"
accelerate launch eval/train_dreambooth_lora_15.py \
--pretrained_model_name_or_path=$MODEL_NAME \
--instance_data_dir=$INSTANCE_DIR \
--output_dir=$OUTPUT_DIR \
--class_data_dir=$CLASS_DIR\
--instance_prompt "an animated painting of a sks person, high quality, masterpiece" \
--class_prompt "an animated painting of a person, high quality, masterpiece" \
--resolution=512 \
--train_batch_size=1 \
--gradient_accumulation_steps=1 \
--learning_rate=1e-4 \
--scale_lr \
--max_train_steps=2000 \
--mixed_precision=bf16 \ |