End of training
Browse files- <v1*>.bin +3 -0
- README.md +22 -0
- attn_maps_color-31_05/output_0_0.png +0 -0
- attn_maps_color-31_05/output_100_0.png +0 -0
- attn_maps_color-31_05/output_150_0.png +0 -0
- attn_maps_color-31_05/output_200_0.png +0 -0
- attn_maps_color-31_05/output_50_0.png +0 -0
- checkpoint-250/custom_checkpoint_0.pkl +3 -0
- checkpoint-250/model.safetensors +3 -0
- checkpoint-250/model_1.safetensors +3 -0
- checkpoint-250/optimizer.bin +3 -0
- checkpoint-250/random_states_0.pkl +3 -0
- checkpoint-250/scheduler.bin +3 -0
- concepts_list.json +1 -0
- logs/custom-diffusion/1716990536.1179004/events.out.tfevents.1716990536.cudahpc06.2358025.1 +3 -0
- logs/custom-diffusion/1716990536.1198945/hparams.yml +58 -0
- logs/custom-diffusion/1716990926.9588747/events.out.tfevents.1716990926.cudahpc06.2424689.1 +3 -0
- logs/custom-diffusion/1716990926.9606037/hparams.yml +58 -0
- logs/custom-diffusion/events.out.tfevents.1716990536.cudahpc06.2358025.0 +3 -0
- logs/custom-diffusion/events.out.tfevents.1716990926.cudahpc06.2424689.0 +3 -0
- pytorch_custom_diffusion_weights.bin +3 -0
- train.sh +61 -0
<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"
|