File size: 2,621 Bytes
4c5e195 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
# export CUDA_VISIBLE_DEVICES=4
# python train_custom_diffusion.py \
# --pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base \
# --num_class_images=200 \
# --resolution=512 \
# --train_batch_size=2 \
# --learning_rate=1e-5 \
# --lr_warmup_steps=0 \
# --max_train_steps=250 \
# --scale_lr \
# --hflip \
# --real_prior \
# --with_prior_preservation \
# --prior_loss_weight=1.0 \
# --class_data_dir "/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/real_reg/samples_dog" \
# --modifier_token "<new1>" \
# --initializer_token "ktn+pll+ucd" \
# --push_to_hub \
# --class_prompt="dog" \
# --prior_loss_weight=0 \
# --instance_prompt="photo of a <new1> dog" \
# --output_dir="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/attn_maps-color_sandesh-dog-clean" \
# --instance_data_dir="/home/xide/miniconda3/code/Attack/mist2-wk/data/single/clean/dog"
# # --instance_data_dir="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/data/single-data/dog" \
export CUDA_VISIBLE_DEVICES=5
# export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/data/Attack/clean/cat"
export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/Attack/mist2-wk/data/single/simply_added/mist_mask_0_background/result_image_100"
export OUTPUT_DIR="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/simply_add/mist_mask_0_background/level_100"
# _prompt_no_cat
# --instance_data_dir="/home/xide/miniconda3/code/data/Attack/clean/cat" \ # clean cat
# --instance_data_dir="/home/xide/miniconda3/code/data/Attack/mist/cat/pgd_8" # mist cat
# --initializer_token "ktn+pll+ucd" \
python train_custom_diffusion.py \
--pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base \
--resolution=512 \
--train_batch_size=2 \
--learning_rate=1e-5 \
--lr_warmup_steps=0 \
--max_train_steps=250 \
--scale_lr \
--hflip \
--modifier_token "<v1*>" \
--initializer_token "ktn" \
--push_to_hub \
--instance_prompt="photo of a <v*> cat" \
--attn_maps_path_name="attn_maps_color-10_6" \
--output_dir=$OUTPUT_DIR \
--instance_data_dir=$INSTANCE_DATA_DIR \
# --real_prior \
# --with_prior_preservation \
# --prior_loss_weight=1.0 \
# --class_data_dir "/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/real_reg/samples_cat" \
# --num_class_images=200 \
# --class_prompt="cat" \
cp "/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/train-single.sh" "$OUTPUT_DIR/train.sh" |