| export CUDA_VISIBLE_DEVICES=4 | |
| # export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/Attack/mist2-wk/data/single/clean/cat" | |
| # export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/Attack/mist2-wk/data/single/clean/person" | |
| # export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/Attack/CAAT/data/1972" | |
| export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/Attack/photoguard/results/complex_attack/test_1_with_prompt/perturbed" | |
| export OUTPUT_DIR="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/complex_attack/test_1_with_prompt" | |
| # export OUTPUT_DIR="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/add_noise_at_latent_levle/random_sample_noise/clean_cat" | |
| # export OUTPUT_DIR="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/debug" | |
| # _prompt_no_cat | |
| # --instance_data_dir="/home/xide/miniconda3/code/Attack/mist2-wk/data/single/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 <v1*> cat" \ | |
| --attn_maps_path_name="attn_maps_color" \ | |
| --output_dir=$OUTPUT_DIR \ | |
| --instance_data_dir=$INSTANCE_DATA_DIR \ | |
| --mist_dir="/home/xide/miniconda3/code/Attack/mist2-wk/data/adv_img/mist/MIST.jpg" \ | |
| --latent_disturb_type="mist" \ | |
| --latent_disturb_eps=0.0078 \ | |
| --adding_mist_befor_vae=False \ | |
| --adding_noise=False \ | |
| # --photoguard=False \ | |
| # --photoguard_mode="" \ # "encoder" | |
| # --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" | |