End of training
Browse files- README.md +6 -0
- checkpoint-10000/controlnet/config.json +57 -0
- checkpoint-10000/controlnet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-10000/optimizer.bin +3 -0
- checkpoint-10000/random_states_0.pkl +3 -0
- checkpoint-10000/scaler.pt +3 -0
- checkpoint-10000/scheduler.bin +3 -0
- checkpoint-15000/controlnet/config.json +57 -0
- checkpoint-15000/controlnet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-15000/optimizer.bin +3 -0
- checkpoint-15000/random_states_0.pkl +3 -0
- checkpoint-15000/scaler.pt +3 -0
- checkpoint-15000/scheduler.bin +3 -0
- checkpoint-20000/controlnet/config.json +57 -0
- checkpoint-20000/controlnet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-20000/optimizer.bin +3 -0
- checkpoint-20000/random_states_0.pkl +3 -0
- checkpoint-20000/scaler.pt +3 -0
- checkpoint-20000/scheduler.bin +3 -0
- checkpoint-25000/controlnet/config.json +57 -0
- checkpoint-25000/controlnet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-25000/optimizer.bin +3 -0
- checkpoint-25000/random_states_0.pkl +3 -0
- checkpoint-25000/scaler.pt +3 -0
- checkpoint-25000/scheduler.bin +3 -0
- checkpoint-5000/controlnet/config.json +57 -0
- checkpoint-5000/controlnet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-5000/optimizer.bin +3 -0
- checkpoint-5000/random_states_0.pkl +3 -0
- checkpoint-5000/scaler.pt +3 -0
- checkpoint-5000/scheduler.bin +3 -0
- diffusion_pytorch_model.safetensors +1 -1
- image_control.png +0 -0
- images_0.png +2 -2
- images_1.png +2 -2
- images_2.png +2 -2
- logs/PosLmkExp/1724202179.9779956/events.out.tfevents.1724202179.workspace-fsi88u3cxa8k-0.6712.1 +3 -0
- logs/PosLmkExp/1724202179.980058/hparams.yml +56 -0
- logs/PosLmkExp/events.out.tfevents.1724202179.workspace-fsi88u3cxa8k-0.6712.0 +3 -0
README.md
CHANGED
@@ -9,6 +9,12 @@ tags:
|
|
9 |
- diffusers
|
10 |
- controlnet
|
11 |
- diffusers-training
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
inference: true
|
13 |
---
|
14 |
|
|
|
9 |
- diffusers
|
10 |
- controlnet
|
11 |
- diffusers-training
|
12 |
+
- stable-diffusion-xl
|
13 |
+
- stable-diffusion-xl-diffusers
|
14 |
+
- text-to-image
|
15 |
+
- diffusers
|
16 |
+
- controlnet
|
17 |
+
- diffusers-training
|
18 |
inference: true
|
19 |
---
|
20 |
|
checkpoint-10000/controlnet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "AlignNet",
|
3 |
+
"_diffusers_version": "0.30.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"addition_embed_type": "text_time",
|
6 |
+
"addition_embed_type_num_heads": 64,
|
7 |
+
"addition_time_embed_dim": 256,
|
8 |
+
"attention_head_dim": [
|
9 |
+
5,
|
10 |
+
10,
|
11 |
+
20
|
12 |
+
],
|
13 |
+
"block_out_channels": [
|
14 |
+
320,
|
15 |
+
640,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"class_embed_type": null,
|
19 |
+
"conditioning_channels": 3,
|
20 |
+
"conditioning_embedding_out_channels": [
|
21 |
+
16,
|
22 |
+
32,
|
23 |
+
96,
|
24 |
+
256
|
25 |
+
],
|
26 |
+
"controlnet_conditioning_channel_order": "rgb",
|
27 |
+
"cross_attention_dim": 2048,
|
28 |
+
"down_block_types": [
|
29 |
+
"DownBlock2D",
|
30 |
+
"CrossAttnDownBlock2D",
|
31 |
+
"CrossAttnDownBlock2D"
|
32 |
+
],
|
33 |
+
"downsample_padding": 1,
|
34 |
+
"encoder_hid_dim": null,
|
35 |
+
"encoder_hid_dim_type": null,
|
36 |
+
"flip_sin_to_cos": true,
|
37 |
+
"freq_shift": 0,
|
38 |
+
"global_pool_conditions": false,
|
39 |
+
"in_channels": 4,
|
40 |
+
"layers_per_block": 2,
|
41 |
+
"mid_block_scale_factor": 1,
|
42 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
43 |
+
"norm_eps": 1e-05,
|
44 |
+
"norm_num_groups": 32,
|
45 |
+
"num_attention_heads": null,
|
46 |
+
"num_class_embeds": null,
|
47 |
+
"only_cross_attention": false,
|
48 |
+
"projection_class_embeddings_input_dim": 2816,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"transformer_layers_per_block": [
|
51 |
+
1,
|
52 |
+
2,
|
53 |
+
10
|
54 |
+
],
|
55 |
+
"upcast_attention": null,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-10000/controlnet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bdfc1461779bbe10eadf2aaa47e550192bfee6cee77d89edb132f226d621c49b
|
3 |
+
size 5042401080
|
checkpoint-10000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5e2a84be66e3813d7af1f428bb401999fb4d6aa9c59d2cfe6a478a0ec95868ed
|
3 |
+
size 10085338361
|
checkpoint-10000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9709c133b2173450f848818a1be485cd295d6ca7ab69a329e911b3ade478e4a6
|
3 |
+
size 15691
|
checkpoint-10000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dd95c18a6b41aa037974831b9a1de45303bee401383908b627a01f41ac3a3109
|
3 |
+
size 557
|
checkpoint-10000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0528393939b459ebeef1788a6bc33c7128c4756bf06d821458a04b69cf42f1b3
|
3 |
+
size 563
|
checkpoint-15000/controlnet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "AlignNet",
|
3 |
+
"_diffusers_version": "0.30.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"addition_embed_type": "text_time",
|
6 |
+
"addition_embed_type_num_heads": 64,
|
7 |
+
"addition_time_embed_dim": 256,
|
8 |
+
"attention_head_dim": [
|
9 |
+
5,
|
10 |
+
10,
|
11 |
+
20
|
12 |
+
],
|
13 |
+
"block_out_channels": [
|
14 |
+
320,
|
15 |
+
640,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"class_embed_type": null,
|
19 |
+
"conditioning_channels": 3,
|
20 |
+
"conditioning_embedding_out_channels": [
|
21 |
+
16,
|
22 |
+
32,
|
23 |
+
96,
|
24 |
+
256
|
25 |
+
],
|
26 |
+
"controlnet_conditioning_channel_order": "rgb",
|
27 |
+
"cross_attention_dim": 2048,
|
28 |
+
"down_block_types": [
|
29 |
+
"DownBlock2D",
|
30 |
+
"CrossAttnDownBlock2D",
|
31 |
+
"CrossAttnDownBlock2D"
|
32 |
+
],
|
33 |
+
"downsample_padding": 1,
|
34 |
+
"encoder_hid_dim": null,
|
35 |
+
"encoder_hid_dim_type": null,
|
36 |
+
"flip_sin_to_cos": true,
|
37 |
+
"freq_shift": 0,
|
38 |
+
"global_pool_conditions": false,
|
39 |
+
"in_channels": 4,
|
40 |
+
"layers_per_block": 2,
|
41 |
+
"mid_block_scale_factor": 1,
|
42 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
43 |
+
"norm_eps": 1e-05,
|
44 |
+
"norm_num_groups": 32,
|
45 |
+
"num_attention_heads": null,
|
46 |
+
"num_class_embeds": null,
|
47 |
+
"only_cross_attention": false,
|
48 |
+
"projection_class_embeddings_input_dim": 2816,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"transformer_layers_per_block": [
|
51 |
+
1,
|
52 |
+
2,
|
53 |
+
10
|
54 |
+
],
|
55 |
+
"upcast_attention": null,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-15000/controlnet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b48ac9cdf73585858c10283d72ff2212d19ba79c7adb5385d7505dbba14df88c
|
3 |
+
size 5042401080
|
checkpoint-15000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:696f2d889ad2960fe20da99e2ca7a9fb6b2db3a78bdc136d1197e7b618a8272c
|
3 |
+
size 10085338361
|
checkpoint-15000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cf1db2841ee78d523caf7b752d09bd17ea953c4f416fa32f142e75f3a5b2eff9
|
3 |
+
size 15691
|
checkpoint-15000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ba90381e07f85bc8f30e20ef1893a232fd0746dfcc8cda103975431a0daca20c
|
3 |
+
size 557
|
checkpoint-15000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:20d36f50c26ec16a3d95e893fd5fef1cf7fedd5212732d3afc2748bcdea5b426
|
3 |
+
size 563
|
checkpoint-20000/controlnet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "AlignNet",
|
3 |
+
"_diffusers_version": "0.30.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"addition_embed_type": "text_time",
|
6 |
+
"addition_embed_type_num_heads": 64,
|
7 |
+
"addition_time_embed_dim": 256,
|
8 |
+
"attention_head_dim": [
|
9 |
+
5,
|
10 |
+
10,
|
11 |
+
20
|
12 |
+
],
|
13 |
+
"block_out_channels": [
|
14 |
+
320,
|
15 |
+
640,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"class_embed_type": null,
|
19 |
+
"conditioning_channels": 3,
|
20 |
+
"conditioning_embedding_out_channels": [
|
21 |
+
16,
|
22 |
+
32,
|
23 |
+
96,
|
24 |
+
256
|
25 |
+
],
|
26 |
+
"controlnet_conditioning_channel_order": "rgb",
|
27 |
+
"cross_attention_dim": 2048,
|
28 |
+
"down_block_types": [
|
29 |
+
"DownBlock2D",
|
30 |
+
"CrossAttnDownBlock2D",
|
31 |
+
"CrossAttnDownBlock2D"
|
32 |
+
],
|
33 |
+
"downsample_padding": 1,
|
34 |
+
"encoder_hid_dim": null,
|
35 |
+
"encoder_hid_dim_type": null,
|
36 |
+
"flip_sin_to_cos": true,
|
37 |
+
"freq_shift": 0,
|
38 |
+
"global_pool_conditions": false,
|
39 |
+
"in_channels": 4,
|
40 |
+
"layers_per_block": 2,
|
41 |
+
"mid_block_scale_factor": 1,
|
42 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
43 |
+
"norm_eps": 1e-05,
|
44 |
+
"norm_num_groups": 32,
|
45 |
+
"num_attention_heads": null,
|
46 |
+
"num_class_embeds": null,
|
47 |
+
"only_cross_attention": false,
|
48 |
+
"projection_class_embeddings_input_dim": 2816,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"transformer_layers_per_block": [
|
51 |
+
1,
|
52 |
+
2,
|
53 |
+
10
|
54 |
+
],
|
55 |
+
"upcast_attention": null,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-20000/controlnet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:28536fea676dea61618c9e5d1d5f5ed3c9dd47612106581f0d2a5e67ef4138c6
|
3 |
+
size 5042401080
|
checkpoint-20000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1f1b90fd87728dc11454b59e88e8b26530f2c7edb08720da823e7a84dc18b400
|
3 |
+
size 10085338361
|
checkpoint-20000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:860bebe7939abb2593744c7744c10de95a9f949a5d187b3da069cca1adf26fc8
|
3 |
+
size 15691
|
checkpoint-20000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9b18447b372bc45953db3e1e3c4a75850bc09a6b743a9a05735c991d11286a91
|
3 |
+
size 557
|
checkpoint-20000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6b0e0b297787b70ab7b1ee342e9452785d218f0e54eed6419e0aed17fff4f43f
|
3 |
+
size 563
|
checkpoint-25000/controlnet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "AlignNet",
|
3 |
+
"_diffusers_version": "0.30.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"addition_embed_type": "text_time",
|
6 |
+
"addition_embed_type_num_heads": 64,
|
7 |
+
"addition_time_embed_dim": 256,
|
8 |
+
"attention_head_dim": [
|
9 |
+
5,
|
10 |
+
10,
|
11 |
+
20
|
12 |
+
],
|
13 |
+
"block_out_channels": [
|
14 |
+
320,
|
15 |
+
640,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"class_embed_type": null,
|
19 |
+
"conditioning_channels": 3,
|
20 |
+
"conditioning_embedding_out_channels": [
|
21 |
+
16,
|
22 |
+
32,
|
23 |
+
96,
|
24 |
+
256
|
25 |
+
],
|
26 |
+
"controlnet_conditioning_channel_order": "rgb",
|
27 |
+
"cross_attention_dim": 2048,
|
28 |
+
"down_block_types": [
|
29 |
+
"DownBlock2D",
|
30 |
+
"CrossAttnDownBlock2D",
|
31 |
+
"CrossAttnDownBlock2D"
|
32 |
+
],
|
33 |
+
"downsample_padding": 1,
|
34 |
+
"encoder_hid_dim": null,
|
35 |
+
"encoder_hid_dim_type": null,
|
36 |
+
"flip_sin_to_cos": true,
|
37 |
+
"freq_shift": 0,
|
38 |
+
"global_pool_conditions": false,
|
39 |
+
"in_channels": 4,
|
40 |
+
"layers_per_block": 2,
|
41 |
+
"mid_block_scale_factor": 1,
|
42 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
43 |
+
"norm_eps": 1e-05,
|
44 |
+
"norm_num_groups": 32,
|
45 |
+
"num_attention_heads": null,
|
46 |
+
"num_class_embeds": null,
|
47 |
+
"only_cross_attention": false,
|
48 |
+
"projection_class_embeddings_input_dim": 2816,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"transformer_layers_per_block": [
|
51 |
+
1,
|
52 |
+
2,
|
53 |
+
10
|
54 |
+
],
|
55 |
+
"upcast_attention": null,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-25000/controlnet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b82c2e85f4b2b3e3908b77920b6e81c6d100a29d22191d0aab3b465b3c1762c4
|
3 |
+
size 5042401080
|
checkpoint-25000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d958f6f0ba19fc5e8c3e38287b3791bd8025ab2eafa0cba5c09bd3508d5e5929
|
3 |
+
size 10085338361
|
checkpoint-25000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c9d96e935e16ead9c0177bbde731c9392503241a8ff98cd41ffe688c5ff23fab
|
3 |
+
size 15691
|
checkpoint-25000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:8ba3823e1ca410d52c677282491617110eb40b2889b87308dd9353318330a377
|
3 |
+
size 557
|
checkpoint-25000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:01686112484ccac88ddb086a9fc888e8b4c7903c83692982d697922eee56fa99
|
3 |
+
size 563
|
checkpoint-5000/controlnet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "AlignNet",
|
3 |
+
"_diffusers_version": "0.30.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"addition_embed_type": "text_time",
|
6 |
+
"addition_embed_type_num_heads": 64,
|
7 |
+
"addition_time_embed_dim": 256,
|
8 |
+
"attention_head_dim": [
|
9 |
+
5,
|
10 |
+
10,
|
11 |
+
20
|
12 |
+
],
|
13 |
+
"block_out_channels": [
|
14 |
+
320,
|
15 |
+
640,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"class_embed_type": null,
|
19 |
+
"conditioning_channels": 3,
|
20 |
+
"conditioning_embedding_out_channels": [
|
21 |
+
16,
|
22 |
+
32,
|
23 |
+
96,
|
24 |
+
256
|
25 |
+
],
|
26 |
+
"controlnet_conditioning_channel_order": "rgb",
|
27 |
+
"cross_attention_dim": 2048,
|
28 |
+
"down_block_types": [
|
29 |
+
"DownBlock2D",
|
30 |
+
"CrossAttnDownBlock2D",
|
31 |
+
"CrossAttnDownBlock2D"
|
32 |
+
],
|
33 |
+
"downsample_padding": 1,
|
34 |
+
"encoder_hid_dim": null,
|
35 |
+
"encoder_hid_dim_type": null,
|
36 |
+
"flip_sin_to_cos": true,
|
37 |
+
"freq_shift": 0,
|
38 |
+
"global_pool_conditions": false,
|
39 |
+
"in_channels": 4,
|
40 |
+
"layers_per_block": 2,
|
41 |
+
"mid_block_scale_factor": 1,
|
42 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
43 |
+
"norm_eps": 1e-05,
|
44 |
+
"norm_num_groups": 32,
|
45 |
+
"num_attention_heads": null,
|
46 |
+
"num_class_embeds": null,
|
47 |
+
"only_cross_attention": false,
|
48 |
+
"projection_class_embeddings_input_dim": 2816,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"transformer_layers_per_block": [
|
51 |
+
1,
|
52 |
+
2,
|
53 |
+
10
|
54 |
+
],
|
55 |
+
"upcast_attention": null,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-5000/controlnet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a0577c916a9c7b06d48b1016bc416ff5d1819e893199d0d5daa8144746edec4f
|
3 |
+
size 5042401080
|
checkpoint-5000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:58121ed5961b971186d7ba02dd3c0d8084ff7f0248df93882c99939a37024c28
|
3 |
+
size 10085338361
|
checkpoint-5000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f557c53e5f37700dc323a9fe5fcf52babbc38462230217c5a68216a7dc1b3e52
|
3 |
+
size 15691
|
checkpoint-5000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a41313b6733b65c1acca917a7c2ced06503c11b898e6cd5de1e6a9f13bfb54ef
|
3 |
+
size 557
|
checkpoint-5000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:83d88135fadf2882eca9ed42328f1c4b7b6df42b91e282d185b8cafa1316f577
|
3 |
+
size 563
|
diffusion_pytorch_model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 5042401080
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:b82c2e85f4b2b3e3908b77920b6e81c6d100a29d22191d0aab3b465b3c1762c4
|
3 |
size 5042401080
|
image_control.png
CHANGED
![]() |
![]() |
images_0.png
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
images_1.png
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
images_2.png
CHANGED
![]() |
Git LFS Details
|
![]() |
Git LFS Details
|
logs/PosLmkExp/1724202179.9779956/events.out.tfevents.1724202179.workspace-fsi88u3cxa8k-0.6712.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:51237d17176e0882d00af3ec1b8c2ed3c3caa907d6c33b7439f54488d185eda5
|
3 |
+
size 2606
|
logs/PosLmkExp/1724202179.980058/hparams.yml
ADDED
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
+
cache_dir: ./cache
|
7 |
+
caption_column: image_caption
|
8 |
+
checkpointing_steps: 5000
|
9 |
+
checkpoints_total_limit: null
|
10 |
+
conditioning_image_column: spiga_seg
|
11 |
+
controlnet_model_name_or_path: null
|
12 |
+
crops_coords_top_left_h: 0
|
13 |
+
crops_coords_top_left_w: 0
|
14 |
+
dataloader_num_workers: 0
|
15 |
+
dataset_config_name: null
|
16 |
+
dataset_name: kwanY/ExpLmkPos
|
17 |
+
enable_npu_flash_attention: false
|
18 |
+
enable_xformers_memory_efficient_attention: true
|
19 |
+
exp_column: image_exp
|
20 |
+
gradient_accumulation_steps: 1
|
21 |
+
gradient_checkpointing: true
|
22 |
+
hub_model_id: null
|
23 |
+
hub_token: null
|
24 |
+
image_column: image
|
25 |
+
learning_rate: 5.0e-06
|
26 |
+
logging_dir: logs
|
27 |
+
lr_num_cycles: 1
|
28 |
+
lr_power: 1.0
|
29 |
+
lr_scheduler: constant
|
30 |
+
lr_warmup_steps: 500
|
31 |
+
max_grad_norm: 1.0
|
32 |
+
max_train_samples: null
|
33 |
+
max_train_steps: 25000
|
34 |
+
mixed_precision: fp16
|
35 |
+
num_train_epochs: 20
|
36 |
+
num_validation_images: 4
|
37 |
+
output_dir: PosLmkExp
|
38 |
+
pose_column: pose
|
39 |
+
pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
|
40 |
+
pretrained_vae_model_name_or_path: null
|
41 |
+
proportion_empty_prompts: 0
|
42 |
+
push_to_hub: true
|
43 |
+
report_to: tensorboard
|
44 |
+
resolution: 512
|
45 |
+
resume_from_checkpoint: null
|
46 |
+
revision: null
|
47 |
+
scale_lr: false
|
48 |
+
seed: null
|
49 |
+
set_grads_to_none: false
|
50 |
+
tokenizer_name: null
|
51 |
+
tracker_project_name: PosLmkExp
|
52 |
+
train_batch_size: 4
|
53 |
+
train_data_dir: null
|
54 |
+
use_8bit_adam: false
|
55 |
+
validation_steps: 2000
|
56 |
+
variant: null
|
logs/PosLmkExp/events.out.tfevents.1724202179.workspace-fsi88u3cxa8k-0.6712.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1a4946c94d0cabc3221acf1d7102eb33dd9472eb17f53d8052d0b326ae923231
|
3 |
+
size 6646801
|