File size: 687 Bytes
ece766c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
export MODEL_NAME="stable-diffusion/stable-diffusion-1-5"
export INSTANCE_DIR="data/training/"
export OUTPUT_DIR="output/mist/"
export CLASS_DIR="data/class"
accelerate launch attacks/mist.py \
--cuda --low_vram_mode --resize \
--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 girl" \
--class_prompt "a oil painting of a girl, high quality, master piece" \
--mixed_precision bf16 \
--max_train_steps 5 \
--checkpointing_iterations 1 \
--prior_loss_weight 0.1 \
--pgd_alpha 0.005 \
--pgd_eps 0.04 \
--max_adv_train_steps 30 \
--max_f_train_steps 10 \ |