# 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/Attack/mist2-wk/data/single/clean/cat" | |
export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/Attack/mist2-wk/data/single/clean/cat" | |
export OUTPUT_DIR="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/add_noise_at_latent_level/mist/mist2latent_vae_10000" | |
# 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-10_6" \ | |
--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="noise" \ | |
--latent_disturb_eps=1.0 | |
# --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" | |