mikephillips
commited on
Commit
•
70e174c
1
Parent(s):
dd8ef83
End of training
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- 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-10000/unet/config.json +57 -0
- checkpoint-10000/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-10000/unet_ema/config.json +64 -0
- checkpoint-10000/unet_ema/diffusion_pytorch_model.bin +3 -0
- checkpoint-12500/optimizer.bin +3 -0
- checkpoint-12500/random_states_0.pkl +3 -0
- checkpoint-12500/scaler.pt +3 -0
- checkpoint-12500/scheduler.bin +3 -0
- checkpoint-12500/unet/config.json +57 -0
- checkpoint-12500/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-12500/unet_ema/config.json +64 -0
- checkpoint-12500/unet_ema/diffusion_pytorch_model.bin +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-15000/unet/config.json +57 -0
- checkpoint-15000/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-15000/unet_ema/config.json +64 -0
- checkpoint-15000/unet_ema/diffusion_pytorch_model.bin +3 -0
- checkpoint-17500/optimizer.bin +3 -0
- checkpoint-17500/random_states_0.pkl +3 -0
- checkpoint-17500/scaler.pt +3 -0
- checkpoint-17500/scheduler.bin +3 -0
- checkpoint-17500/unet/config.json +57 -0
- checkpoint-17500/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-17500/unet_ema/config.json +64 -0
- checkpoint-17500/unet_ema/diffusion_pytorch_model.bin +3 -0
- checkpoint-2500/optimizer.bin +3 -0
- checkpoint-2500/random_states_0.pkl +3 -0
- checkpoint-2500/scaler.pt +3 -0
- checkpoint-2500/scheduler.bin +3 -0
- checkpoint-2500/unet/config.json +57 -0
- checkpoint-2500/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-2500/unet_ema/config.json +64 -0
- checkpoint-2500/unet_ema/diffusion_pytorch_model.bin +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
- checkpoint-5000/unet/config.json +57 -0
- checkpoint-5000/unet/diffusion_pytorch_model.bin +3 -0
- checkpoint-5000/unet_ema/config.json +64 -0
- checkpoint-5000/unet_ema/diffusion_pytorch_model.bin +3 -0
- checkpoint-7500/optimizer.bin +3 -0
- checkpoint-7500/random_states_0.pkl +3 -0
checkpoint-10000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ba28c07fc9ec998429a33cad17ccfd5a076b30e1fe11bf22a0a984d4a0dc3dcc
|
3 |
+
size 6927866927
|
checkpoint-10000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2a811c86facb1333be037d41eb971bff49aaad62711d0d1b9f3d1b57c3cc5bf1
|
3 |
+
size 15691
|
checkpoint-10000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
|
3 |
+
size 557
|
checkpoint-10000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9f865fb658c50c3ef7ff79d832c3c900c9c60ab0fe13cfb1c7ce868a88c81ec6
|
3 |
+
size 563
|
checkpoint-10000/unet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"down_block_types": [
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"DownBlock2D"
|
29 |
+
],
|
30 |
+
"downsample_padding": 1,
|
31 |
+
"dual_cross_attention": false,
|
32 |
+
"flip_sin_to_cos": true,
|
33 |
+
"freq_shift": 0,
|
34 |
+
"in_channels": 4,
|
35 |
+
"layers_per_block": 2,
|
36 |
+
"mid_block_scale_factor": 1,
|
37 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
38 |
+
"norm_eps": 1e-05,
|
39 |
+
"norm_num_groups": 32,
|
40 |
+
"num_class_embeds": null,
|
41 |
+
"only_cross_attention": false,
|
42 |
+
"out_channels": 4,
|
43 |
+
"projection_class_embeddings_input_dim": null,
|
44 |
+
"resnet_time_scale_shift": "default",
|
45 |
+
"sample_size": 64,
|
46 |
+
"time_cond_proj_dim": null,
|
47 |
+
"time_embedding_type": "positional",
|
48 |
+
"timestep_post_act": null,
|
49 |
+
"up_block_types": [
|
50 |
+
"UpBlock2D",
|
51 |
+
"CrossAttnUpBlock2D",
|
52 |
+
"CrossAttnUpBlock2D",
|
53 |
+
"CrossAttnUpBlock2D"
|
54 |
+
],
|
55 |
+
"upcast_attention": false,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-10000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:163a3b78cd35119b024eeb7f625cfd506fd521d8d5a1e9078444f9a4cbc94aaf
|
3 |
+
size 3463934693
|
checkpoint-10000/unet_ema/config.json
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"decay": 0.9999,
|
25 |
+
"down_block_types": [
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"CrossAttnDownBlock2D",
|
29 |
+
"DownBlock2D"
|
30 |
+
],
|
31 |
+
"downsample_padding": 1,
|
32 |
+
"dual_cross_attention": false,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"inv_gamma": 1.0,
|
37 |
+
"layers_per_block": 2,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"min_decay": 0.0,
|
41 |
+
"norm_eps": 1e-05,
|
42 |
+
"norm_num_groups": 32,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"optimization_step": 10000,
|
46 |
+
"out_channels": 4,
|
47 |
+
"power": 0.6666666666666666,
|
48 |
+
"projection_class_embeddings_input_dim": null,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_type": "positional",
|
53 |
+
"timestep_post_act": null,
|
54 |
+
"up_block_types": [
|
55 |
+
"UpBlock2D",
|
56 |
+
"CrossAttnUpBlock2D",
|
57 |
+
"CrossAttnUpBlock2D",
|
58 |
+
"CrossAttnUpBlock2D"
|
59 |
+
],
|
60 |
+
"upcast_attention": false,
|
61 |
+
"update_after_step": 0,
|
62 |
+
"use_ema_warmup": false,
|
63 |
+
"use_linear_projection": true
|
64 |
+
}
|
checkpoint-10000/unet_ema/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0ec51f383a6cacdb35bc67dd662735ca45419b170d7d8c76f79f2ada24c113e6
|
3 |
+
size 3463925093
|
checkpoint-12500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a35061f6207a159eb68840253e14cd71eddd0fa567e4f2cec5e7ca977a4a8cb1
|
3 |
+
size 6927866927
|
checkpoint-12500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ba75c63851b082bd38d90e984f35035274494689f81807a8104f9a20fb532744
|
3 |
+
size 15691
|
checkpoint-12500/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
|
3 |
+
size 557
|
checkpoint-12500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:199f985fb5053a67baaa5e8d12ed3ed7ea83d13b5f5e6e62bba27336dee0e893
|
3 |
+
size 563
|
checkpoint-12500/unet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"down_block_types": [
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"DownBlock2D"
|
29 |
+
],
|
30 |
+
"downsample_padding": 1,
|
31 |
+
"dual_cross_attention": false,
|
32 |
+
"flip_sin_to_cos": true,
|
33 |
+
"freq_shift": 0,
|
34 |
+
"in_channels": 4,
|
35 |
+
"layers_per_block": 2,
|
36 |
+
"mid_block_scale_factor": 1,
|
37 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
38 |
+
"norm_eps": 1e-05,
|
39 |
+
"norm_num_groups": 32,
|
40 |
+
"num_class_embeds": null,
|
41 |
+
"only_cross_attention": false,
|
42 |
+
"out_channels": 4,
|
43 |
+
"projection_class_embeddings_input_dim": null,
|
44 |
+
"resnet_time_scale_shift": "default",
|
45 |
+
"sample_size": 64,
|
46 |
+
"time_cond_proj_dim": null,
|
47 |
+
"time_embedding_type": "positional",
|
48 |
+
"timestep_post_act": null,
|
49 |
+
"up_block_types": [
|
50 |
+
"UpBlock2D",
|
51 |
+
"CrossAttnUpBlock2D",
|
52 |
+
"CrossAttnUpBlock2D",
|
53 |
+
"CrossAttnUpBlock2D"
|
54 |
+
],
|
55 |
+
"upcast_attention": false,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-12500/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:163a3b78cd35119b024eeb7f625cfd506fd521d8d5a1e9078444f9a4cbc94aaf
|
3 |
+
size 3463934693
|
checkpoint-12500/unet_ema/config.json
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"decay": 0.9999,
|
25 |
+
"down_block_types": [
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"CrossAttnDownBlock2D",
|
29 |
+
"DownBlock2D"
|
30 |
+
],
|
31 |
+
"downsample_padding": 1,
|
32 |
+
"dual_cross_attention": false,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"inv_gamma": 1.0,
|
37 |
+
"layers_per_block": 2,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"min_decay": 0.0,
|
41 |
+
"norm_eps": 1e-05,
|
42 |
+
"norm_num_groups": 32,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"optimization_step": 12500,
|
46 |
+
"out_channels": 4,
|
47 |
+
"power": 0.6666666666666666,
|
48 |
+
"projection_class_embeddings_input_dim": null,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_type": "positional",
|
53 |
+
"timestep_post_act": null,
|
54 |
+
"up_block_types": [
|
55 |
+
"UpBlock2D",
|
56 |
+
"CrossAttnUpBlock2D",
|
57 |
+
"CrossAttnUpBlock2D",
|
58 |
+
"CrossAttnUpBlock2D"
|
59 |
+
],
|
60 |
+
"upcast_attention": false,
|
61 |
+
"update_after_step": 0,
|
62 |
+
"use_ema_warmup": false,
|
63 |
+
"use_linear_projection": true
|
64 |
+
}
|
checkpoint-12500/unet_ema/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0ec51f383a6cacdb35bc67dd662735ca45419b170d7d8c76f79f2ada24c113e6
|
3 |
+
size 3463925093
|
checkpoint-15000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1b254db9dc5f0488547e3937624e4efabc46b2b5190ea65d9f7859be7ab01734
|
3 |
+
size 6927866927
|
checkpoint-15000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ebfa015892f94cb2d301f32773afe38306f10a5473f3060b81a0c64992f7d214
|
3 |
+
size 15691
|
checkpoint-15000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
|
3 |
+
size 557
|
checkpoint-15000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:f06e63d66851390129c0bc8dfd30ac909baee42bed87fe004c5752dda0c5b164
|
3 |
+
size 563
|
checkpoint-15000/unet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"down_block_types": [
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"DownBlock2D"
|
29 |
+
],
|
30 |
+
"downsample_padding": 1,
|
31 |
+
"dual_cross_attention": false,
|
32 |
+
"flip_sin_to_cos": true,
|
33 |
+
"freq_shift": 0,
|
34 |
+
"in_channels": 4,
|
35 |
+
"layers_per_block": 2,
|
36 |
+
"mid_block_scale_factor": 1,
|
37 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
38 |
+
"norm_eps": 1e-05,
|
39 |
+
"norm_num_groups": 32,
|
40 |
+
"num_class_embeds": null,
|
41 |
+
"only_cross_attention": false,
|
42 |
+
"out_channels": 4,
|
43 |
+
"projection_class_embeddings_input_dim": null,
|
44 |
+
"resnet_time_scale_shift": "default",
|
45 |
+
"sample_size": 64,
|
46 |
+
"time_cond_proj_dim": null,
|
47 |
+
"time_embedding_type": "positional",
|
48 |
+
"timestep_post_act": null,
|
49 |
+
"up_block_types": [
|
50 |
+
"UpBlock2D",
|
51 |
+
"CrossAttnUpBlock2D",
|
52 |
+
"CrossAttnUpBlock2D",
|
53 |
+
"CrossAttnUpBlock2D"
|
54 |
+
],
|
55 |
+
"upcast_attention": false,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-15000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:163a3b78cd35119b024eeb7f625cfd506fd521d8d5a1e9078444f9a4cbc94aaf
|
3 |
+
size 3463934693
|
checkpoint-15000/unet_ema/config.json
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"decay": 0.9999,
|
25 |
+
"down_block_types": [
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"CrossAttnDownBlock2D",
|
29 |
+
"DownBlock2D"
|
30 |
+
],
|
31 |
+
"downsample_padding": 1,
|
32 |
+
"dual_cross_attention": false,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"inv_gamma": 1.0,
|
37 |
+
"layers_per_block": 2,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"min_decay": 0.0,
|
41 |
+
"norm_eps": 1e-05,
|
42 |
+
"norm_num_groups": 32,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"optimization_step": 15000,
|
46 |
+
"out_channels": 4,
|
47 |
+
"power": 0.6666666666666666,
|
48 |
+
"projection_class_embeddings_input_dim": null,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_type": "positional",
|
53 |
+
"timestep_post_act": null,
|
54 |
+
"up_block_types": [
|
55 |
+
"UpBlock2D",
|
56 |
+
"CrossAttnUpBlock2D",
|
57 |
+
"CrossAttnUpBlock2D",
|
58 |
+
"CrossAttnUpBlock2D"
|
59 |
+
],
|
60 |
+
"upcast_attention": false,
|
61 |
+
"update_after_step": 0,
|
62 |
+
"use_ema_warmup": false,
|
63 |
+
"use_linear_projection": true
|
64 |
+
}
|
checkpoint-15000/unet_ema/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0ec51f383a6cacdb35bc67dd662735ca45419b170d7d8c76f79f2ada24c113e6
|
3 |
+
size 3463925093
|
checkpoint-17500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:967309eb3758fe8f7d4634d8dbe61529e1ff4b953a2a10d563490f60f52a2286
|
3 |
+
size 6927866927
|
checkpoint-17500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3a04db4859700b58cc24ce19c3bedf0cc3923bc53b39a5a0a9ccd3bcd78a99a9
|
3 |
+
size 15691
|
checkpoint-17500/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
|
3 |
+
size 557
|
checkpoint-17500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fbea3a38c85784bdbc626d6f3788c8e5d4414c42f41152ae068e3e7b9aa2217b
|
3 |
+
size 563
|
checkpoint-17500/unet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"down_block_types": [
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"DownBlock2D"
|
29 |
+
],
|
30 |
+
"downsample_padding": 1,
|
31 |
+
"dual_cross_attention": false,
|
32 |
+
"flip_sin_to_cos": true,
|
33 |
+
"freq_shift": 0,
|
34 |
+
"in_channels": 4,
|
35 |
+
"layers_per_block": 2,
|
36 |
+
"mid_block_scale_factor": 1,
|
37 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
38 |
+
"norm_eps": 1e-05,
|
39 |
+
"norm_num_groups": 32,
|
40 |
+
"num_class_embeds": null,
|
41 |
+
"only_cross_attention": false,
|
42 |
+
"out_channels": 4,
|
43 |
+
"projection_class_embeddings_input_dim": null,
|
44 |
+
"resnet_time_scale_shift": "default",
|
45 |
+
"sample_size": 64,
|
46 |
+
"time_cond_proj_dim": null,
|
47 |
+
"time_embedding_type": "positional",
|
48 |
+
"timestep_post_act": null,
|
49 |
+
"up_block_types": [
|
50 |
+
"UpBlock2D",
|
51 |
+
"CrossAttnUpBlock2D",
|
52 |
+
"CrossAttnUpBlock2D",
|
53 |
+
"CrossAttnUpBlock2D"
|
54 |
+
],
|
55 |
+
"upcast_attention": false,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-17500/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:163a3b78cd35119b024eeb7f625cfd506fd521d8d5a1e9078444f9a4cbc94aaf
|
3 |
+
size 3463934693
|
checkpoint-17500/unet_ema/config.json
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"decay": 0.9999,
|
25 |
+
"down_block_types": [
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"CrossAttnDownBlock2D",
|
29 |
+
"DownBlock2D"
|
30 |
+
],
|
31 |
+
"downsample_padding": 1,
|
32 |
+
"dual_cross_attention": false,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"inv_gamma": 1.0,
|
37 |
+
"layers_per_block": 2,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"min_decay": 0.0,
|
41 |
+
"norm_eps": 1e-05,
|
42 |
+
"norm_num_groups": 32,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"optimization_step": 17500,
|
46 |
+
"out_channels": 4,
|
47 |
+
"power": 0.6666666666666666,
|
48 |
+
"projection_class_embeddings_input_dim": null,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_type": "positional",
|
53 |
+
"timestep_post_act": null,
|
54 |
+
"up_block_types": [
|
55 |
+
"UpBlock2D",
|
56 |
+
"CrossAttnUpBlock2D",
|
57 |
+
"CrossAttnUpBlock2D",
|
58 |
+
"CrossAttnUpBlock2D"
|
59 |
+
],
|
60 |
+
"upcast_attention": false,
|
61 |
+
"update_after_step": 0,
|
62 |
+
"use_ema_warmup": false,
|
63 |
+
"use_linear_projection": true
|
64 |
+
}
|
checkpoint-17500/unet_ema/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0ec51f383a6cacdb35bc67dd662735ca45419b170d7d8c76f79f2ada24c113e6
|
3 |
+
size 3463925093
|
checkpoint-2500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4fc642481cbc843a986f589a9d0639b6429444c7c9873d610a3c173191bd5ca4
|
3 |
+
size 6927866927
|
checkpoint-2500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:91238b7b00be26899116534fa75d668d2d8b09cbbbaad4cf8ce1953e56ae9e3b
|
3 |
+
size 15691
|
checkpoint-2500/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
|
3 |
+
size 557
|
checkpoint-2500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ca36524046558f2fe52e139053f4e7352aeb344eb5c2f51c4581c6013043139c
|
3 |
+
size 563
|
checkpoint-2500/unet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"down_block_types": [
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"DownBlock2D"
|
29 |
+
],
|
30 |
+
"downsample_padding": 1,
|
31 |
+
"dual_cross_attention": false,
|
32 |
+
"flip_sin_to_cos": true,
|
33 |
+
"freq_shift": 0,
|
34 |
+
"in_channels": 4,
|
35 |
+
"layers_per_block": 2,
|
36 |
+
"mid_block_scale_factor": 1,
|
37 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
38 |
+
"norm_eps": 1e-05,
|
39 |
+
"norm_num_groups": 32,
|
40 |
+
"num_class_embeds": null,
|
41 |
+
"only_cross_attention": false,
|
42 |
+
"out_channels": 4,
|
43 |
+
"projection_class_embeddings_input_dim": null,
|
44 |
+
"resnet_time_scale_shift": "default",
|
45 |
+
"sample_size": 64,
|
46 |
+
"time_cond_proj_dim": null,
|
47 |
+
"time_embedding_type": "positional",
|
48 |
+
"timestep_post_act": null,
|
49 |
+
"up_block_types": [
|
50 |
+
"UpBlock2D",
|
51 |
+
"CrossAttnUpBlock2D",
|
52 |
+
"CrossAttnUpBlock2D",
|
53 |
+
"CrossAttnUpBlock2D"
|
54 |
+
],
|
55 |
+
"upcast_attention": false,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-2500/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:163a3b78cd35119b024eeb7f625cfd506fd521d8d5a1e9078444f9a4cbc94aaf
|
3 |
+
size 3463934693
|
checkpoint-2500/unet_ema/config.json
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"decay": 0.9999,
|
25 |
+
"down_block_types": [
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"CrossAttnDownBlock2D",
|
29 |
+
"DownBlock2D"
|
30 |
+
],
|
31 |
+
"downsample_padding": 1,
|
32 |
+
"dual_cross_attention": false,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"inv_gamma": 1.0,
|
37 |
+
"layers_per_block": 2,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"min_decay": 0.0,
|
41 |
+
"norm_eps": 1e-05,
|
42 |
+
"norm_num_groups": 32,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"optimization_step": 2500,
|
46 |
+
"out_channels": 4,
|
47 |
+
"power": 0.6666666666666666,
|
48 |
+
"projection_class_embeddings_input_dim": null,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_type": "positional",
|
53 |
+
"timestep_post_act": null,
|
54 |
+
"up_block_types": [
|
55 |
+
"UpBlock2D",
|
56 |
+
"CrossAttnUpBlock2D",
|
57 |
+
"CrossAttnUpBlock2D",
|
58 |
+
"CrossAttnUpBlock2D"
|
59 |
+
],
|
60 |
+
"upcast_attention": false,
|
61 |
+
"update_after_step": 0,
|
62 |
+
"use_ema_warmup": false,
|
63 |
+
"use_linear_projection": true
|
64 |
+
}
|
checkpoint-2500/unet_ema/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0ec51f383a6cacdb35bc67dd662735ca45419b170d7d8c76f79f2ada24c113e6
|
3 |
+
size 3463925093
|
checkpoint-5000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ffd02ec8a7070df2042b15374806630de081ecb56a7e7d76a68e2a7ce215e436
|
3 |
+
size 6927866927
|
checkpoint-5000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:daad2679d9f6db408fcc64cd873683089d933c6ca1a72ca5d293803b5f17846b
|
3 |
+
size 15691
|
checkpoint-5000/scaler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7fa181fa360d46feed4180ea17c8b6a4a879a9b4231c2e91aff2be20be9076cc
|
3 |
+
size 557
|
checkpoint-5000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:c87d873cd01fa0efe110cd060290c21e143d90b0bfb4ec455eef39e5bfae96b5
|
3 |
+
size 563
|
checkpoint-5000/unet/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"down_block_types": [
|
25 |
+
"CrossAttnDownBlock2D",
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"DownBlock2D"
|
29 |
+
],
|
30 |
+
"downsample_padding": 1,
|
31 |
+
"dual_cross_attention": false,
|
32 |
+
"flip_sin_to_cos": true,
|
33 |
+
"freq_shift": 0,
|
34 |
+
"in_channels": 4,
|
35 |
+
"layers_per_block": 2,
|
36 |
+
"mid_block_scale_factor": 1,
|
37 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
38 |
+
"norm_eps": 1e-05,
|
39 |
+
"norm_num_groups": 32,
|
40 |
+
"num_class_embeds": null,
|
41 |
+
"only_cross_attention": false,
|
42 |
+
"out_channels": 4,
|
43 |
+
"projection_class_embeddings_input_dim": null,
|
44 |
+
"resnet_time_scale_shift": "default",
|
45 |
+
"sample_size": 64,
|
46 |
+
"time_cond_proj_dim": null,
|
47 |
+
"time_embedding_type": "positional",
|
48 |
+
"timestep_post_act": null,
|
49 |
+
"up_block_types": [
|
50 |
+
"UpBlock2D",
|
51 |
+
"CrossAttnUpBlock2D",
|
52 |
+
"CrossAttnUpBlock2D",
|
53 |
+
"CrossAttnUpBlock2D"
|
54 |
+
],
|
55 |
+
"upcast_attention": false,
|
56 |
+
"use_linear_projection": true
|
57 |
+
}
|
checkpoint-5000/unet/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:163a3b78cd35119b024eeb7f625cfd506fd521d8d5a1e9078444f9a4cbc94aaf
|
3 |
+
size 3463934693
|
checkpoint-5000/unet_ema/config.json
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DConditionModel",
|
3 |
+
"_diffusers_version": "0.15.0.dev0",
|
4 |
+
"_name_or_path": "stabilityai/stable-diffusion-2-1-base",
|
5 |
+
"act_fn": "silu",
|
6 |
+
"attention_head_dim": [
|
7 |
+
5,
|
8 |
+
10,
|
9 |
+
20,
|
10 |
+
20
|
11 |
+
],
|
12 |
+
"block_out_channels": [
|
13 |
+
320,
|
14 |
+
640,
|
15 |
+
1280,
|
16 |
+
1280
|
17 |
+
],
|
18 |
+
"center_input_sample": false,
|
19 |
+
"class_embed_type": null,
|
20 |
+
"class_embeddings_concat": false,
|
21 |
+
"conv_in_kernel": 3,
|
22 |
+
"conv_out_kernel": 3,
|
23 |
+
"cross_attention_dim": 1024,
|
24 |
+
"decay": 0.9999,
|
25 |
+
"down_block_types": [
|
26 |
+
"CrossAttnDownBlock2D",
|
27 |
+
"CrossAttnDownBlock2D",
|
28 |
+
"CrossAttnDownBlock2D",
|
29 |
+
"DownBlock2D"
|
30 |
+
],
|
31 |
+
"downsample_padding": 1,
|
32 |
+
"dual_cross_attention": false,
|
33 |
+
"flip_sin_to_cos": true,
|
34 |
+
"freq_shift": 0,
|
35 |
+
"in_channels": 4,
|
36 |
+
"inv_gamma": 1.0,
|
37 |
+
"layers_per_block": 2,
|
38 |
+
"mid_block_scale_factor": 1,
|
39 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
40 |
+
"min_decay": 0.0,
|
41 |
+
"norm_eps": 1e-05,
|
42 |
+
"norm_num_groups": 32,
|
43 |
+
"num_class_embeds": null,
|
44 |
+
"only_cross_attention": false,
|
45 |
+
"optimization_step": 5000,
|
46 |
+
"out_channels": 4,
|
47 |
+
"power": 0.6666666666666666,
|
48 |
+
"projection_class_embeddings_input_dim": null,
|
49 |
+
"resnet_time_scale_shift": "default",
|
50 |
+
"sample_size": 64,
|
51 |
+
"time_cond_proj_dim": null,
|
52 |
+
"time_embedding_type": "positional",
|
53 |
+
"timestep_post_act": null,
|
54 |
+
"up_block_types": [
|
55 |
+
"UpBlock2D",
|
56 |
+
"CrossAttnUpBlock2D",
|
57 |
+
"CrossAttnUpBlock2D",
|
58 |
+
"CrossAttnUpBlock2D"
|
59 |
+
],
|
60 |
+
"upcast_attention": false,
|
61 |
+
"update_after_step": 0,
|
62 |
+
"use_ema_warmup": false,
|
63 |
+
"use_linear_projection": true
|
64 |
+
}
|
checkpoint-5000/unet_ema/diffusion_pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0ec51f383a6cacdb35bc67dd662735ca45419b170d7d8c76f79f2ada24c113e6
|
3 |
+
size 3463925093
|
checkpoint-7500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1f450cf96dacf1be3ae714eadf0988177c8a44abcde6db2523ba2b5a0fa4b990
|
3 |
+
size 6927866927
|
checkpoint-7500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1681ba00c02eb82a9a83b08f27f86f3432fedac0745c99047f766eaf533dca17
|
3 |
+
size 15691
|