End of training
Browse files- <new1>.bin +3 -0
- <new2>.bin +3 -0
- README.md +22 -0
- checkpoint-250/custom_checkpoint_0.pkl +3 -0
- checkpoint-250/optimizer.bin +3 -0
- checkpoint-250/pytorch_model.bin +3 -0
- checkpoint-250/pytorch_model_1.bin +3 -0
- checkpoint-250/random_states_0.pkl +3 -0
- checkpoint-250/scheduler.bin +3 -0
- checkpoint-500/custom_checkpoint_0.pkl +3 -0
- checkpoint-500/optimizer.bin +3 -0
- checkpoint-500/pytorch_model.bin +3 -0
- checkpoint-500/pytorch_model_1.bin +3 -0
- checkpoint-500/random_states_0.pkl +3 -0
- checkpoint-500/scheduler.bin +3 -0
- logs/custom-diffusion/1701364511.2446833/events.out.tfevents.1701364511.cudahpc06.2125436.1 +3 -0
- logs/custom-diffusion/1701364511.2467477/hparams.yml +57 -0
- logs/custom-diffusion/1701364583.2795281/events.out.tfevents.1701364583.cudahpc06.2125621.1 +3 -0
- logs/custom-diffusion/1701364583.281561/hparams.yml +57 -0
- logs/custom-diffusion/1701364654.5909536/events.out.tfevents.1701364654.cudahpc06.2125815.1 +3 -0
- logs/custom-diffusion/1701364654.5925028/hparams.yml +57 -0
- logs/custom-diffusion/1701364758.5754466/events.out.tfevents.1701364758.cudahpc06.2126147.1 +3 -0
- logs/custom-diffusion/1701364758.577048/hparams.yml +57 -0
- logs/custom-diffusion/1701364809.4193866/events.out.tfevents.1701364809.cudahpc06.2126591.1 +3 -0
- logs/custom-diffusion/1701364809.421006/hparams.yml +57 -0
- logs/custom-diffusion/events.out.tfevents.1701364511.cudahpc06.2125436.0 +3 -0
- logs/custom-diffusion/events.out.tfevents.1701364583.cudahpc06.2125621.0 +3 -0
- logs/custom-diffusion/events.out.tfevents.1701364654.cudahpc06.2125815.0 +3 -0
- logs/custom-diffusion/events.out.tfevents.1701364758.cudahpc06.2126147.0 +3 -0
- logs/custom-diffusion/events.out.tfevents.1701364809.cudahpc06.2126591.0 +3 -0
- pytorch_custom_diffusion_weights.bin +3 -0
<new1>.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ff9d83c751bee0706a85c5a31c4f6c394834818c821a3ca94ce7f8078078f4cc
|
3 |
+
size 151788200
|
<new2>.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b6b7df3952e2e9e655788fc1b40b2611f760582a2159903d1903afc841e6a054
|
3 |
+
size 151788200
|
README.md
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
---
|
3 |
+
license: creativeml-openrail-m
|
4 |
+
base_model: CompVis/stable-diffusion-v1-4
|
5 |
+
instance_prompt: None
|
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/path-to-save-model-multi
|
16 |
+
|
17 |
+
These are Custom Diffusion adaption weights for CompVis/stable-diffusion-v1-4. The weights were trained on None 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).
|
checkpoint-250/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23b7e3e3773afeae3eb2936aa429240b371d79eed929546e4c6467cb15f0ac76
|
3 |
+
size 76689565
|
checkpoint-250/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:af7ac0394b7390e75877a6b533fc0036f8ce6a1509ea91e7f3f86e8ed9817515
|
3 |
+
size 456956998
|
checkpoint-250/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f25186247b2546b68f19f77f70c19d04e63c7366b014975f89187f70b6398377
|
3 |
+
size 76689361
|
checkpoint-250/pytorch_model_1.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1a40559fb81d63d0c47c248eda21a929cbde363ef2eac6883156481cc013f842
|
3 |
+
size 492324905
|
checkpoint-250/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8e297aecbc13ad8a87194636500618d2f5dacc87838d74895fdee9d67fb8dc3c
|
3 |
+
size 21731
|
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
|
checkpoint-500/custom_checkpoint_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ce07cd22053beb711b75ea68ab7097dd4d9789226d7a88ab84f832b7473837af
|
3 |
+
size 76689565
|
checkpoint-500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:60cd155a49a174f5f4c1b4f2e6c8b332db32a4c9d6d2ce60b2902e47e6b8befe
|
3 |
+
size 456956998
|
checkpoint-500/pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c6d8a36c8fc402e0cddf88d93667e6e6c56eea8175175460bd0242b82079ddfe
|
3 |
+
size 76689361
|
checkpoint-500/pytorch_model_1.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9d35bb84918ae724f213d2fbf6a972533603a27b48ba11fb84e1ce0e9d0456b8
|
3 |
+
size 492324905
|
checkpoint-500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cdf3de2c2f59ade90c561705a0a5f342c573859a2ed1cd773ee2db4f47905407
|
3 |
+
size 21731
|
checkpoint-500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1834aa974518fe3f070ca2e5278d1335ab53153a59c8a3352af3906f194ce477
|
3 |
+
size 563
|
logs/custom-diffusion/1701364511.2446833/events.out.tfevents.1701364511.cudahpc06.2125436.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8326709324f82d03c4a03dcaa504978de7da167cc5d7b06dc7ee3036637448f3
|
3 |
+
size 2620
|
logs/custom-diffusion/1701364511.2467477/hparams.yml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
center_crop: false
|
7 |
+
checkpointing_steps: 250
|
8 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_prompt: null
|
11 |
+
concepts_list: ./examples/custom_diffusion/concept_list.json
|
12 |
+
dataloader_num_workers: 2
|
13 |
+
enable_xformers_memory_efficient_attention: false
|
14 |
+
freeze_model: crossattn_kv
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hflip: true
|
18 |
+
hub_model_id: null
|
19 |
+
hub_token: null
|
20 |
+
initializer_token: ktn+pll+ucd
|
21 |
+
instance_data_dir: null
|
22 |
+
instance_prompt: null
|
23 |
+
learning_rate: 1.0e-05
|
24 |
+
local_rank: -1
|
25 |
+
logging_dir: logs
|
26 |
+
lr_scheduler: constant
|
27 |
+
lr_warmup_steps: 0
|
28 |
+
max_grad_norm: 1.0
|
29 |
+
max_train_steps: 1250
|
30 |
+
mixed_precision: null
|
31 |
+
modifier_token: <new1>+<new2>+<new3>+<new4>+<new5>
|
32 |
+
no_safe_serialization: true
|
33 |
+
noaug: false
|
34 |
+
num_class_images: 200
|
35 |
+
num_train_epochs: 1
|
36 |
+
num_validation_images: 2
|
37 |
+
output_dir: path-to-save-model-multi
|
38 |
+
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
|
39 |
+
prior_generation_precision: null
|
40 |
+
prior_loss_weight: 0.0
|
41 |
+
push_to_hub: true
|
42 |
+
real_prior: false
|
43 |
+
report_to: tensorboard
|
44 |
+
resolution: 512
|
45 |
+
resume_from_checkpoint: null
|
46 |
+
revision: null
|
47 |
+
sample_batch_size: 4
|
48 |
+
scale_lr: true
|
49 |
+
seed: 42
|
50 |
+
set_grads_to_none: false
|
51 |
+
tokenizer_name: null
|
52 |
+
train_batch_size: 2
|
53 |
+
use_8bit_adam: false
|
54 |
+
validation_prompt: null
|
55 |
+
validation_steps: 50
|
56 |
+
variant: null
|
57 |
+
with_prior_preservation: false
|
logs/custom-diffusion/1701364583.2795281/events.out.tfevents.1701364583.cudahpc06.2125621.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1ba8198c085a44b32d21e4a19dfb6a9d2a2a4cd5cc1d1143fb327e8ca270fd89
|
3 |
+
size 2594
|
logs/custom-diffusion/1701364583.281561/hparams.yml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
center_crop: false
|
7 |
+
checkpointing_steps: 250
|
8 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_prompt: null
|
11 |
+
concepts_list: ./concept_list.json
|
12 |
+
dataloader_num_workers: 2
|
13 |
+
enable_xformers_memory_efficient_attention: false
|
14 |
+
freeze_model: crossattn_kv
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hflip: true
|
18 |
+
hub_model_id: null
|
19 |
+
hub_token: null
|
20 |
+
initializer_token: ktn+pll+ucd
|
21 |
+
instance_data_dir: null
|
22 |
+
instance_prompt: null
|
23 |
+
learning_rate: 1.0e-05
|
24 |
+
local_rank: -1
|
25 |
+
logging_dir: logs
|
26 |
+
lr_scheduler: constant
|
27 |
+
lr_warmup_steps: 0
|
28 |
+
max_grad_norm: 1.0
|
29 |
+
max_train_steps: 1250
|
30 |
+
mixed_precision: null
|
31 |
+
modifier_token: <new1>+<new2>+<new3>+<new4>+<new5>
|
32 |
+
no_safe_serialization: true
|
33 |
+
noaug: false
|
34 |
+
num_class_images: 200
|
35 |
+
num_train_epochs: 1
|
36 |
+
num_validation_images: 2
|
37 |
+
output_dir: path-to-save-model-multi
|
38 |
+
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
|
39 |
+
prior_generation_precision: null
|
40 |
+
prior_loss_weight: 0.0
|
41 |
+
push_to_hub: true
|
42 |
+
real_prior: false
|
43 |
+
report_to: tensorboard
|
44 |
+
resolution: 512
|
45 |
+
resume_from_checkpoint: null
|
46 |
+
revision: null
|
47 |
+
sample_batch_size: 4
|
48 |
+
scale_lr: true
|
49 |
+
seed: 42
|
50 |
+
set_grads_to_none: false
|
51 |
+
tokenizer_name: null
|
52 |
+
train_batch_size: 2
|
53 |
+
use_8bit_adam: false
|
54 |
+
validation_prompt: null
|
55 |
+
validation_steps: 50
|
56 |
+
variant: null
|
57 |
+
with_prior_preservation: false
|
logs/custom-diffusion/1701364654.5909536/events.out.tfevents.1701364654.cudahpc06.2125815.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9a10c88323a73185644457e191b178d8454e7eb5ec63461aa0ae7570e58dbb43
|
3 |
+
size 2594
|
logs/custom-diffusion/1701364654.5925028/hparams.yml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
center_crop: false
|
7 |
+
checkpointing_steps: 250
|
8 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_prompt: null
|
11 |
+
concepts_list: ./concept_list.json
|
12 |
+
dataloader_num_workers: 2
|
13 |
+
enable_xformers_memory_efficient_attention: false
|
14 |
+
freeze_model: crossattn_kv
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hflip: true
|
18 |
+
hub_model_id: null
|
19 |
+
hub_token: null
|
20 |
+
initializer_token: ktn+pll+ucd
|
21 |
+
instance_data_dir: null
|
22 |
+
instance_prompt: null
|
23 |
+
learning_rate: 1.0e-05
|
24 |
+
local_rank: -1
|
25 |
+
logging_dir: logs
|
26 |
+
lr_scheduler: constant
|
27 |
+
lr_warmup_steps: 0
|
28 |
+
max_grad_norm: 1.0
|
29 |
+
max_train_steps: 1250
|
30 |
+
mixed_precision: null
|
31 |
+
modifier_token: <new1>+<new2>+<new3>+<new4>+<new5>
|
32 |
+
no_safe_serialization: true
|
33 |
+
noaug: false
|
34 |
+
num_class_images: 200
|
35 |
+
num_train_epochs: 1
|
36 |
+
num_validation_images: 2
|
37 |
+
output_dir: path-to-save-model-multi
|
38 |
+
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
|
39 |
+
prior_generation_precision: null
|
40 |
+
prior_loss_weight: 0.0
|
41 |
+
push_to_hub: true
|
42 |
+
real_prior: false
|
43 |
+
report_to: tensorboard
|
44 |
+
resolution: 512
|
45 |
+
resume_from_checkpoint: null
|
46 |
+
revision: null
|
47 |
+
sample_batch_size: 4
|
48 |
+
scale_lr: true
|
49 |
+
seed: 42
|
50 |
+
set_grads_to_none: false
|
51 |
+
tokenizer_name: null
|
52 |
+
train_batch_size: 2
|
53 |
+
use_8bit_adam: false
|
54 |
+
validation_prompt: null
|
55 |
+
validation_steps: 50
|
56 |
+
variant: null
|
57 |
+
with_prior_preservation: false
|
logs/custom-diffusion/1701364758.5754466/events.out.tfevents.1701364758.cudahpc06.2126147.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bbad95071768d4fa1c3fb2544288cbc015b05de869897cc5eb4039719f0cd3f7
|
3 |
+
size 2573
|
logs/custom-diffusion/1701364758.577048/hparams.yml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
center_crop: false
|
7 |
+
checkpointing_steps: 250
|
8 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_prompt: null
|
11 |
+
concepts_list: ./concept_list.json
|
12 |
+
dataloader_num_workers: 2
|
13 |
+
enable_xformers_memory_efficient_attention: false
|
14 |
+
freeze_model: crossattn_kv
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hflip: true
|
18 |
+
hub_model_id: null
|
19 |
+
hub_token: null
|
20 |
+
initializer_token: ktn+pll+ucd
|
21 |
+
instance_data_dir: null
|
22 |
+
instance_prompt: null
|
23 |
+
learning_rate: 1.0e-05
|
24 |
+
local_rank: -1
|
25 |
+
logging_dir: logs
|
26 |
+
lr_scheduler: constant
|
27 |
+
lr_warmup_steps: 0
|
28 |
+
max_grad_norm: 1.0
|
29 |
+
max_train_steps: 1250
|
30 |
+
mixed_precision: null
|
31 |
+
modifier_token: <new1>+<new2>
|
32 |
+
no_safe_serialization: true
|
33 |
+
noaug: false
|
34 |
+
num_class_images: 200
|
35 |
+
num_train_epochs: 1
|
36 |
+
num_validation_images: 2
|
37 |
+
output_dir: path-to-save-model-multi
|
38 |
+
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
|
39 |
+
prior_generation_precision: null
|
40 |
+
prior_loss_weight: 0.0
|
41 |
+
push_to_hub: true
|
42 |
+
real_prior: false
|
43 |
+
report_to: tensorboard
|
44 |
+
resolution: 512
|
45 |
+
resume_from_checkpoint: null
|
46 |
+
revision: null
|
47 |
+
sample_batch_size: 4
|
48 |
+
scale_lr: true
|
49 |
+
seed: 42
|
50 |
+
set_grads_to_none: false
|
51 |
+
tokenizer_name: null
|
52 |
+
train_batch_size: 2
|
53 |
+
use_8bit_adam: false
|
54 |
+
validation_prompt: null
|
55 |
+
validation_steps: 50
|
56 |
+
variant: null
|
57 |
+
with_prior_preservation: false
|
logs/custom-diffusion/1701364809.4193866/events.out.tfevents.1701364809.cudahpc06.2126591.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e81d0816893abe4bb60f0ecb7d98b5b3de48e435a5485f7f9237bed79f46a90c
|
3 |
+
size 2573
|
logs/custom-diffusion/1701364809.421006/hparams.yml
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
center_crop: false
|
7 |
+
checkpointing_steps: 250
|
8 |
+
checkpoints_total_limit: null
|
9 |
+
class_data_dir: null
|
10 |
+
class_prompt: null
|
11 |
+
concepts_list: ./concept_list.json
|
12 |
+
dataloader_num_workers: 2
|
13 |
+
enable_xformers_memory_efficient_attention: false
|
14 |
+
freeze_model: crossattn_kv
|
15 |
+
gradient_accumulation_steps: 1
|
16 |
+
gradient_checkpointing: false
|
17 |
+
hflip: true
|
18 |
+
hub_model_id: null
|
19 |
+
hub_token: null
|
20 |
+
initializer_token: ktn+pll+ucd
|
21 |
+
instance_data_dir: null
|
22 |
+
instance_prompt: null
|
23 |
+
learning_rate: 1.0e-05
|
24 |
+
local_rank: -1
|
25 |
+
logging_dir: logs
|
26 |
+
lr_scheduler: constant
|
27 |
+
lr_warmup_steps: 0
|
28 |
+
max_grad_norm: 1.0
|
29 |
+
max_train_steps: 500
|
30 |
+
mixed_precision: null
|
31 |
+
modifier_token: <new1>+<new2>
|
32 |
+
no_safe_serialization: true
|
33 |
+
noaug: false
|
34 |
+
num_class_images: 200
|
35 |
+
num_train_epochs: 1
|
36 |
+
num_validation_images: 2
|
37 |
+
output_dir: path-to-save-model-multi
|
38 |
+
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
|
39 |
+
prior_generation_precision: null
|
40 |
+
prior_loss_weight: 0.0
|
41 |
+
push_to_hub: true
|
42 |
+
real_prior: false
|
43 |
+
report_to: tensorboard
|
44 |
+
resolution: 512
|
45 |
+
resume_from_checkpoint: null
|
46 |
+
revision: null
|
47 |
+
sample_batch_size: 4
|
48 |
+
scale_lr: true
|
49 |
+
seed: 42
|
50 |
+
set_grads_to_none: false
|
51 |
+
tokenizer_name: null
|
52 |
+
train_batch_size: 2
|
53 |
+
use_8bit_adam: false
|
54 |
+
validation_prompt: null
|
55 |
+
validation_steps: 50
|
56 |
+
variant: null
|
57 |
+
with_prior_preservation: false
|
logs/custom-diffusion/events.out.tfevents.1701364511.cudahpc06.2125436.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:41f035d0f724a11f96f0e1defadea72c66bbb7142395e99e00dbb415b4d3e492
|
3 |
+
size 88
|
logs/custom-diffusion/events.out.tfevents.1701364583.cudahpc06.2125621.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f86deabb8fd8cf5b644a093e085b783d394b36a43ef6fa163c295dc0e98febb6
|
3 |
+
size 88
|
logs/custom-diffusion/events.out.tfevents.1701364654.cudahpc06.2125815.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:34f7d8006a04506845b35787268ac4ffcae4088cd9ff8dfcd129e153115b6155
|
3 |
+
size 88
|
logs/custom-diffusion/events.out.tfevents.1701364758.cudahpc06.2126147.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ee4cce7e6b0f75214bb61a0b590790fbe3c3fe817d09645d3e1ef78c83860b76
|
3 |
+
size 826
|
logs/custom-diffusion/events.out.tfevents.1701364809.cudahpc06.2126591.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:502a92935c5b68fcec61779cb6cfa591f84f7d513e200d163c7cd2769df7985f
|
3 |
+
size 41834
|
pytorch_custom_diffusion_weights.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:af98073afe82bd3a2708f164acf692da7adc14af497c3362120ad8da1d9ce2ce
|
3 |
+
size 76691351
|