Epoch 0
Browse files- .gitignore +2 -0
- checkpoint-1000/optimizer.bin +3 -0
- checkpoint-1000/random_states_0.pkl +3 -0
- checkpoint-1000/scheduler.bin +3 -0
- checkpoint-1000/unet/config.json +50 -0
- checkpoint-1000/unet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-1000/unet_ema/config.json +57 -0
- checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-500/optimizer.bin +3 -0
- checkpoint-500/random_states_0.pkl +3 -0
- checkpoint-500/scheduler.bin +3 -0
- checkpoint-500/unet/config.json +50 -0
- checkpoint-500/unet/diffusion_pytorch_model.safetensors +3 -0
- checkpoint-500/unet_ema/config.json +57 -0
- checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors +3 -0
- logs/train_unconditional/events.out.tfevents.1696507938.jupiter.2833.0 +3 -0
- model_index.json +12 -0
- scheduler/scheduler_config.json +18 -0
- unet/config.json +50 -0
- unet/diffusion_pytorch_model.safetensors +3 -0
.gitignore
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
step_*
|
2 |
+
epoch_*
|
checkpoint-1000/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4f2a20553f0665d4cb5b84e1f06a70507bf33ced765b857685802e1bc857a84f
|
3 |
+
size 909760147
|
checkpoint-1000/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0406c8e5091a0d341f9d190c42f748a3561abe27a39593a6e47a09c15bb0dc7c
|
3 |
+
size 17627
|
checkpoint-1000/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ea8f6e3e6c28c0ce11622b6de1eb53eb86d06d3ffdc2e39493e0530ec588ff45
|
3 |
+
size 559
|
checkpoint-1000/unet/config.json
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DModel",
|
3 |
+
"_diffusers_version": "0.22.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"add_attention": true,
|
6 |
+
"attention_head_dim": 8,
|
7 |
+
"attn_norm_num_groups": null,
|
8 |
+
"block_out_channels": [
|
9 |
+
128,
|
10 |
+
128,
|
11 |
+
256,
|
12 |
+
256,
|
13 |
+
512,
|
14 |
+
512
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"down_block_types": [
|
19 |
+
"DownBlock2D",
|
20 |
+
"DownBlock2D",
|
21 |
+
"DownBlock2D",
|
22 |
+
"DownBlock2D",
|
23 |
+
"AttnDownBlock2D",
|
24 |
+
"DownBlock2D"
|
25 |
+
],
|
26 |
+
"downsample_padding": 1,
|
27 |
+
"downsample_type": "conv",
|
28 |
+
"dropout": 0.0,
|
29 |
+
"flip_sin_to_cos": true,
|
30 |
+
"freq_shift": 0,
|
31 |
+
"in_channels": 3,
|
32 |
+
"layers_per_block": 2,
|
33 |
+
"mid_block_scale_factor": 1,
|
34 |
+
"norm_eps": 1e-05,
|
35 |
+
"norm_num_groups": 32,
|
36 |
+
"num_class_embeds": null,
|
37 |
+
"out_channels": 3,
|
38 |
+
"resnet_time_scale_shift": "default",
|
39 |
+
"sample_size": 256,
|
40 |
+
"time_embedding_type": "positional",
|
41 |
+
"up_block_types": [
|
42 |
+
"UpBlock2D",
|
43 |
+
"AttnUpBlock2D",
|
44 |
+
"UpBlock2D",
|
45 |
+
"UpBlock2D",
|
46 |
+
"UpBlock2D",
|
47 |
+
"UpBlock2D"
|
48 |
+
],
|
49 |
+
"upsample_type": "conv"
|
50 |
+
}
|
checkpoint-1000/unet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:23676f8da821b935d0ad221780018ef046bb439b7ef95395db0b11c72fdd502a
|
3 |
+
size 454741108
|
checkpoint-1000/unet_ema/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DModel",
|
3 |
+
"_diffusers_version": "0.22.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"add_attention": true,
|
6 |
+
"attention_head_dim": 8,
|
7 |
+
"attn_norm_num_groups": null,
|
8 |
+
"block_out_channels": [
|
9 |
+
128,
|
10 |
+
128,
|
11 |
+
256,
|
12 |
+
256,
|
13 |
+
512,
|
14 |
+
512
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"decay": 0.9999,
|
19 |
+
"down_block_types": [
|
20 |
+
"DownBlock2D",
|
21 |
+
"DownBlock2D",
|
22 |
+
"DownBlock2D",
|
23 |
+
"DownBlock2D",
|
24 |
+
"AttnDownBlock2D",
|
25 |
+
"DownBlock2D"
|
26 |
+
],
|
27 |
+
"downsample_padding": 1,
|
28 |
+
"downsample_type": "conv",
|
29 |
+
"dropout": 0.0,
|
30 |
+
"flip_sin_to_cos": true,
|
31 |
+
"freq_shift": 0,
|
32 |
+
"in_channels": 3,
|
33 |
+
"inv_gamma": 1.0,
|
34 |
+
"layers_per_block": 2,
|
35 |
+
"mid_block_scale_factor": 1,
|
36 |
+
"min_decay": 0.0,
|
37 |
+
"norm_eps": 1e-05,
|
38 |
+
"norm_num_groups": 32,
|
39 |
+
"num_class_embeds": null,
|
40 |
+
"optimization_step": 1000,
|
41 |
+
"out_channels": 3,
|
42 |
+
"power": 0.75,
|
43 |
+
"resnet_time_scale_shift": "default",
|
44 |
+
"sample_size": 256,
|
45 |
+
"time_embedding_type": "positional",
|
46 |
+
"up_block_types": [
|
47 |
+
"UpBlock2D",
|
48 |
+
"AttnUpBlock2D",
|
49 |
+
"UpBlock2D",
|
50 |
+
"UpBlock2D",
|
51 |
+
"UpBlock2D",
|
52 |
+
"UpBlock2D"
|
53 |
+
],
|
54 |
+
"update_after_step": 0,
|
55 |
+
"upsample_type": "conv",
|
56 |
+
"use_ema_warmup": true
|
57 |
+
}
|
checkpoint-1000/unet_ema/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:cf048a1987dcea4aea4a55c532ef6ca85e051bf80f5f493e815ab68a7adb15bc
|
3 |
+
size 454741108
|
checkpoint-500/optimizer.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:787091de931743af99ab238053d74d89e788556d271cf512240903d745a7153a
|
3 |
+
size 909760147
|
checkpoint-500/random_states_0.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4dc62e32cf8699de92714ddbd322e2d382285b12ca5799f4b027f5d4451661e7
|
3 |
+
size 17627
|
checkpoint-500/scheduler.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:4ad24ac5ea5c3202e60fdb8bae1423958d8f8767386931a78a197b50b83c1719
|
3 |
+
size 559
|
checkpoint-500/unet/config.json
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DModel",
|
3 |
+
"_diffusers_version": "0.22.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"add_attention": true,
|
6 |
+
"attention_head_dim": 8,
|
7 |
+
"attn_norm_num_groups": null,
|
8 |
+
"block_out_channels": [
|
9 |
+
128,
|
10 |
+
128,
|
11 |
+
256,
|
12 |
+
256,
|
13 |
+
512,
|
14 |
+
512
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"down_block_types": [
|
19 |
+
"DownBlock2D",
|
20 |
+
"DownBlock2D",
|
21 |
+
"DownBlock2D",
|
22 |
+
"DownBlock2D",
|
23 |
+
"AttnDownBlock2D",
|
24 |
+
"DownBlock2D"
|
25 |
+
],
|
26 |
+
"downsample_padding": 1,
|
27 |
+
"downsample_type": "conv",
|
28 |
+
"dropout": 0.0,
|
29 |
+
"flip_sin_to_cos": true,
|
30 |
+
"freq_shift": 0,
|
31 |
+
"in_channels": 3,
|
32 |
+
"layers_per_block": 2,
|
33 |
+
"mid_block_scale_factor": 1,
|
34 |
+
"norm_eps": 1e-05,
|
35 |
+
"norm_num_groups": 32,
|
36 |
+
"num_class_embeds": null,
|
37 |
+
"out_channels": 3,
|
38 |
+
"resnet_time_scale_shift": "default",
|
39 |
+
"sample_size": 256,
|
40 |
+
"time_embedding_type": "positional",
|
41 |
+
"up_block_types": [
|
42 |
+
"UpBlock2D",
|
43 |
+
"AttnUpBlock2D",
|
44 |
+
"UpBlock2D",
|
45 |
+
"UpBlock2D",
|
46 |
+
"UpBlock2D",
|
47 |
+
"UpBlock2D"
|
48 |
+
],
|
49 |
+
"upsample_type": "conv"
|
50 |
+
}
|
checkpoint-500/unet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fd68a40fb995c99669a0b5a42cda485789e3a7346d81711384ebd81c6002c2cf
|
3 |
+
size 454741108
|
checkpoint-500/unet_ema/config.json
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DModel",
|
3 |
+
"_diffusers_version": "0.22.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"add_attention": true,
|
6 |
+
"attention_head_dim": 8,
|
7 |
+
"attn_norm_num_groups": null,
|
8 |
+
"block_out_channels": [
|
9 |
+
128,
|
10 |
+
128,
|
11 |
+
256,
|
12 |
+
256,
|
13 |
+
512,
|
14 |
+
512
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"decay": 0.9999,
|
19 |
+
"down_block_types": [
|
20 |
+
"DownBlock2D",
|
21 |
+
"DownBlock2D",
|
22 |
+
"DownBlock2D",
|
23 |
+
"DownBlock2D",
|
24 |
+
"AttnDownBlock2D",
|
25 |
+
"DownBlock2D"
|
26 |
+
],
|
27 |
+
"downsample_padding": 1,
|
28 |
+
"downsample_type": "conv",
|
29 |
+
"dropout": 0.0,
|
30 |
+
"flip_sin_to_cos": true,
|
31 |
+
"freq_shift": 0,
|
32 |
+
"in_channels": 3,
|
33 |
+
"inv_gamma": 1.0,
|
34 |
+
"layers_per_block": 2,
|
35 |
+
"mid_block_scale_factor": 1,
|
36 |
+
"min_decay": 0.0,
|
37 |
+
"norm_eps": 1e-05,
|
38 |
+
"norm_num_groups": 32,
|
39 |
+
"num_class_embeds": null,
|
40 |
+
"optimization_step": 500,
|
41 |
+
"out_channels": 3,
|
42 |
+
"power": 0.75,
|
43 |
+
"resnet_time_scale_shift": "default",
|
44 |
+
"sample_size": 256,
|
45 |
+
"time_embedding_type": "positional",
|
46 |
+
"up_block_types": [
|
47 |
+
"UpBlock2D",
|
48 |
+
"AttnUpBlock2D",
|
49 |
+
"UpBlock2D",
|
50 |
+
"UpBlock2D",
|
51 |
+
"UpBlock2D",
|
52 |
+
"UpBlock2D"
|
53 |
+
],
|
54 |
+
"update_after_step": 0,
|
55 |
+
"upsample_type": "conv",
|
56 |
+
"use_ema_warmup": true
|
57 |
+
}
|
checkpoint-500/unet_ema/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5284fe79261c8d2f4d31612976b1c314fcf94f7118a61399885d34baf92477b9
|
3 |
+
size 454741108
|
logs/train_unconditional/events.out.tfevents.1696507938.jupiter.2833.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6f32cfba428d5821e503fb8ed8c62428560bb2809abe1db7b52f6313a54d2e32
|
3 |
+
size 1796039
|
model_index.json
ADDED
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "DDPMPipeline",
|
3 |
+
"_diffusers_version": "0.22.0.dev0",
|
4 |
+
"scheduler": [
|
5 |
+
"diffusers",
|
6 |
+
"DDPMScheduler"
|
7 |
+
],
|
8 |
+
"unet": [
|
9 |
+
"diffusers",
|
10 |
+
"UNet2DModel"
|
11 |
+
]
|
12 |
+
}
|
scheduler/scheduler_config.json
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "DDPMScheduler",
|
3 |
+
"_diffusers_version": "0.22.0.dev0",
|
4 |
+
"beta_end": 0.02,
|
5 |
+
"beta_schedule": "linear",
|
6 |
+
"beta_start": 0.0001,
|
7 |
+
"clip_sample": true,
|
8 |
+
"clip_sample_range": 1.0,
|
9 |
+
"dynamic_thresholding_ratio": 0.995,
|
10 |
+
"num_train_timesteps": 1000,
|
11 |
+
"prediction_type": "epsilon",
|
12 |
+
"sample_max_value": 1.0,
|
13 |
+
"steps_offset": 0,
|
14 |
+
"thresholding": false,
|
15 |
+
"timestep_spacing": "leading",
|
16 |
+
"trained_betas": null,
|
17 |
+
"variance_type": "fixed_small"
|
18 |
+
}
|
unet/config.json
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_class_name": "UNet2DModel",
|
3 |
+
"_diffusers_version": "0.22.0.dev0",
|
4 |
+
"act_fn": "silu",
|
5 |
+
"add_attention": true,
|
6 |
+
"attention_head_dim": 8,
|
7 |
+
"attn_norm_num_groups": null,
|
8 |
+
"block_out_channels": [
|
9 |
+
128,
|
10 |
+
128,
|
11 |
+
256,
|
12 |
+
256,
|
13 |
+
512,
|
14 |
+
512
|
15 |
+
],
|
16 |
+
"center_input_sample": false,
|
17 |
+
"class_embed_type": null,
|
18 |
+
"down_block_types": [
|
19 |
+
"DownBlock2D",
|
20 |
+
"DownBlock2D",
|
21 |
+
"DownBlock2D",
|
22 |
+
"DownBlock2D",
|
23 |
+
"AttnDownBlock2D",
|
24 |
+
"DownBlock2D"
|
25 |
+
],
|
26 |
+
"downsample_padding": 1,
|
27 |
+
"downsample_type": "conv",
|
28 |
+
"dropout": 0.0,
|
29 |
+
"flip_sin_to_cos": true,
|
30 |
+
"freq_shift": 0,
|
31 |
+
"in_channels": 3,
|
32 |
+
"layers_per_block": 2,
|
33 |
+
"mid_block_scale_factor": 1,
|
34 |
+
"norm_eps": 1e-05,
|
35 |
+
"norm_num_groups": 32,
|
36 |
+
"num_class_embeds": null,
|
37 |
+
"out_channels": 3,
|
38 |
+
"resnet_time_scale_shift": "default",
|
39 |
+
"sample_size": 256,
|
40 |
+
"time_embedding_type": "positional",
|
41 |
+
"up_block_types": [
|
42 |
+
"UpBlock2D",
|
43 |
+
"AttnUpBlock2D",
|
44 |
+
"UpBlock2D",
|
45 |
+
"UpBlock2D",
|
46 |
+
"UpBlock2D",
|
47 |
+
"UpBlock2D"
|
48 |
+
],
|
49 |
+
"upsample_type": "conv"
|
50 |
+
}
|
unet/diffusion_pytorch_model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0cb23820e464488d24e163f6a1f938618476f0c09824276f4e002cd0e314406b
|
3 |
+
size 454741108
|