| # 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" |