SidXXD commited on
Commit
cf078f5
1 Parent(s): 49dba5a

End of training

Browse files
<v1*>.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a48c2dfc8a1c20841731718f684dbbb18eccdf1ab4f18a1c69449e3fd6327c9a
3
+ size 202379941
README.md ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ license: creativeml-openrail-m
4
+ base_model: stabilityai/stable-diffusion-2-1-base
5
+ instance_prompt: <v1*> a photo of cat
6
+ tags:
7
+ - stable-diffusion
8
+ - stable-diffusion-diffusers
9
+ - text-to-image
10
+ - diffusers
11
+ - custom-diffusion
12
+ inference: true
13
+ ---
14
+
15
+ # Custom Diffusion - SidXXD/mist_8-v_a_photo_of_cat-token_init_cat
16
+
17
+ These are Custom Diffusion adaption weights for stabilityai/stable-diffusion-2-1-base. The weights were trained on <v1*> a photo of cat using [Custom Diffusion](https://www.cs.cmu.edu/~custom-diffusion). You can find some example images in the following.
18
+
19
+
20
+
21
+
22
+ For more details on the training, please follow [this link](https://github.com/huggingface/diffusers/blob/main/examples/custom_diffusion).
attn_maps_color-31_05/output_0_0.png ADDED
attn_maps_color-31_05/output_100_0.png ADDED
attn_maps_color-31_05/output_150_0.png ADDED
attn_maps_color-31_05/output_200_0.png ADDED
attn_maps_color-31_05/output_50_0.png ADDED
checkpoint-250/custom_checkpoint_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aca49aac7bd272129bb3706cbd0e78f79076d384e9437a7d6285be451648c3f3
3
+ size 102248605
checkpoint-250/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:260572039ff62e3e93d871d2e3a4c07f1edd35990603b98687882eeb4d5b1407
3
+ size 102241440
checkpoint-250/model_1.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ed98e80528121ac41736f58610f1d2924811fa0bbb8ab79ff5dcec23befbcef
3
+ size 1361600400
checkpoint-250/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:680d40c1a69fef9c2be17258b1c099ae245fe2d3a424e2f8643b74027e8f8f87
3
+ size 609258566
checkpoint-250/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ebc30de0291c4dfd98d70db9279bb226cb8ab9fd8682df18f3e74cf6c05db8fc
3
+ size 14663
checkpoint-250/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c54c7dfa0981e4f3405702dde32f7c139b056257f5c201760ea6fbfb47a4b1d7
3
+ size 563
concepts_list.json ADDED
@@ -0,0 +1 @@
 
 
1
+ [{"instance_prompt": "<v1*> a photo of cat", "class_prompt": null, "instance_data_dir": "/home/xide/miniconda3/code/data/Attack/mist/cat/pgd_8", "class_data_dir": null}]
logs/custom-diffusion/1716990536.1179004/events.out.tfevents.1716990536.cudahpc06.2358025.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f73081fe2d82358fc8fb013cd97efce5000cacf7bed567195a335ee345d64117
3
+ size 2861
logs/custom-diffusion/1716990536.1198945/hparams.yml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ attn_maps_path_name: attn_maps_color-31_05
7
+ center_crop: false
8
+ checkpointing_steps: 250
9
+ checkpoints_total_limit: null
10
+ class_data_dir: null
11
+ class_prompt: null
12
+ concepts_list: null
13
+ dataloader_num_workers: 2
14
+ enable_xformers_memory_efficient_attention: false
15
+ freeze_model: crossattn_kv
16
+ gradient_accumulation_steps: 1
17
+ gradient_checkpointing: false
18
+ hflip: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ initializer_token: cat
22
+ instance_data_dir: /home/xide/miniconda3/code/data/Attack/mist/cat/pgd_8
23
+ instance_prompt: <v1*> a photo of cat
24
+ learning_rate: 1.0e-05
25
+ local_rank: -1
26
+ logging_dir: logs
27
+ lr_scheduler: constant
28
+ lr_warmup_steps: 0
29
+ max_grad_norm: 1.0
30
+ max_train_steps: 250
31
+ mixed_precision: null
32
+ modifier_token: <v1*>
33
+ no_safe_serialization: true
34
+ noaug: false
35
+ num_class_images: 200
36
+ num_train_epochs: 1
37
+ num_validation_images: 2
38
+ output_dir: /home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/attn_maps/dif_position_v_token/mist_8-v_a_photo_of_cat-token_init_cat
39
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1-base
40
+ prior_generation_precision: null
41
+ prior_loss_weight: 0
42
+ push_to_hub: true
43
+ real_prior: false
44
+ report_to: tensorboard
45
+ resolution: 512
46
+ resume_from_checkpoint: null
47
+ revision: null
48
+ sample_batch_size: 4
49
+ scale_lr: true
50
+ seed: 42
51
+ set_grads_to_none: false
52
+ tokenizer_name: null
53
+ train_batch_size: 2
54
+ use_8bit_adam: false
55
+ validation_prompt: null
56
+ validation_steps: 50
57
+ variant: null
58
+ with_prior_preservation: false
logs/custom-diffusion/1716990926.9588747/events.out.tfevents.1716990926.cudahpc06.2424689.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e7d6240527feb33a927a9c7d5102c62c1fa6a1dcb8e958871fb769b3d8db71b
3
+ size 2861
logs/custom-diffusion/1716990926.9606037/hparams.yml ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ attn_maps_path_name: attn_maps_color-31_05
7
+ center_crop: false
8
+ checkpointing_steps: 250
9
+ checkpoints_total_limit: null
10
+ class_data_dir: null
11
+ class_prompt: null
12
+ concepts_list: null
13
+ dataloader_num_workers: 2
14
+ enable_xformers_memory_efficient_attention: false
15
+ freeze_model: crossattn_kv
16
+ gradient_accumulation_steps: 1
17
+ gradient_checkpointing: false
18
+ hflip: true
19
+ hub_model_id: null
20
+ hub_token: null
21
+ initializer_token: cat
22
+ instance_data_dir: /home/xide/miniconda3/code/data/Attack/mist/cat/pgd_8
23
+ instance_prompt: <v1*> a photo of cat
24
+ learning_rate: 1.0e-05
25
+ local_rank: -1
26
+ logging_dir: logs
27
+ lr_scheduler: constant
28
+ lr_warmup_steps: 0
29
+ max_grad_norm: 1.0
30
+ max_train_steps: 250
31
+ mixed_precision: null
32
+ modifier_token: <v1*>
33
+ no_safe_serialization: true
34
+ noaug: false
35
+ num_class_images: 200
36
+ num_train_epochs: 1
37
+ num_validation_images: 2
38
+ output_dir: /home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/attn_maps/dif_position_v_token/mist_8-v_a_photo_of_cat-token_init_cat
39
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1-base
40
+ prior_generation_precision: null
41
+ prior_loss_weight: 0
42
+ push_to_hub: true
43
+ real_prior: false
44
+ report_to: tensorboard
45
+ resolution: 512
46
+ resume_from_checkpoint: null
47
+ revision: null
48
+ sample_batch_size: 4
49
+ scale_lr: true
50
+ seed: 42
51
+ set_grads_to_none: false
52
+ tokenizer_name: null
53
+ train_batch_size: 2
54
+ use_8bit_adam: false
55
+ validation_prompt: null
56
+ validation_steps: 50
57
+ variant: null
58
+ with_prior_preservation: false
logs/custom-diffusion/events.out.tfevents.1716990536.cudahpc06.2358025.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d8e247aa8a77f59f30597070e3898ff609dbe940032a87fb3149edf37755aab
3
+ size 19826
logs/custom-diffusion/events.out.tfevents.1716990926.cudahpc06.2424689.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9d6464bccfccc4c90b16b3b11d22cafc9ff5800d0b2b802a6690537819752fc8
3
+ size 20834
pytorch_custom_diffusion_weights.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d973641348f29cfcba0eb2dee817f06651883a5aa62a1eba45fbaf2998007309
3
+ size 102250391
train.sh ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # export CUDA_VISIBLE_DEVICES=4
2
+
3
+ # python train_custom_diffusion.py \
4
+ # --pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base \
5
+ # --num_class_images=200 \
6
+ # --resolution=512 \
7
+ # --train_batch_size=2 \
8
+ # --learning_rate=1e-5 \
9
+ # --lr_warmup_steps=0 \
10
+ # --max_train_steps=250 \
11
+ # --scale_lr \
12
+ # --hflip \
13
+ # --real_prior \
14
+ # --with_prior_preservation \
15
+ # --prior_loss_weight=1.0 \
16
+ # --class_data_dir "/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/real_reg/samples_dog" \
17
+ # --modifier_token "<new1>" \
18
+ # --initializer_token "ktn+pll+ucd" \
19
+ # --push_to_hub \
20
+ # --class_prompt="dog" \
21
+ # --prior_loss_weight=0 \
22
+ # --instance_prompt="photo of a <new1> dog" \
23
+ # --output_dir="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/attn_maps-color_sandesh-dog-clean" \
24
+ # --instance_data_dir="/home/xide/miniconda3/code/Attack/mist2-wk/data/single/clean/dog"
25
+ # # --instance_data_dir="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/data/single-data/dog" \
26
+
27
+
28
+ export CUDA_VISIBLE_DEVICES=6
29
+ export INSTANCE_DATA_DIR="/home/xide/miniconda3/code/data/Attack/mist/cat/pgd_8"
30
+ export OUTPUT_DIR="/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/model/attn_maps/dif_position_v_token/mist_8-v_a_photo_of_cat-token_init_cat"
31
+ # _prompt_no_cat
32
+
33
+ # --instance_data_dir="/home/xide/miniconda3/code/data/Attack/clean/cat" \ # clean cat
34
+ # --instance_data_dir="/home/xide/miniconda3/code/data/Attack/mist/cat/pgd_8" # mist cat
35
+ # --initializer_token "ktn+pll+ucd" \
36
+
37
+ python train_custom_diffusion.py \
38
+ --pretrained_model_name_or_path=stabilityai/stable-diffusion-2-1-base \
39
+ --resolution=512 \
40
+ --train_batch_size=2 \
41
+ --learning_rate=1e-5 \
42
+ --lr_warmup_steps=0 \
43
+ --max_train_steps=250 \
44
+ --scale_lr \
45
+ --hflip \
46
+ --modifier_token "<v1*>" \
47
+ --initializer_token "cat" \
48
+ --push_to_hub \
49
+ --instance_prompt="<v1*> a photo of cat" \
50
+ --attn_maps_path_name="attn_maps_color-31_05" \
51
+ --output_dir=$OUTPUT_DIR \
52
+ --instance_data_dir=$INSTANCE_DATA_DIR \
53
+ # --real_prior \
54
+ # --with_prior_preservation \
55
+ # --prior_loss_weight=1.0 \
56
+ # --class_data_dir "/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/real_reg/samples_cat" \
57
+ # --num_class_images=200 \
58
+ # --class_prompt="cat" \
59
+
60
+
61
+ cp "/home/xide/miniconda3/code/diffusers/examples/custom_diffusion/train-single.sh" "$OUTPUT_DIR/train.sh"